Skip to content

Commit

Permalink
Disable CW in manual script due to profiling issue (#1818)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #1818

Disable Columnwise Sharding in manual script as profiling takes a massive amount of time (result is a huge profile). This will not affect the automatic benchmark

Differential Revision: D55173220

fbshipit-source-id: c1bbfafeeda49ceb2a8ab224931247b357ea8fd6
  • Loading branch information
PaulZhang12 authored and facebook-github-bot committed Mar 21, 2024
1 parent 19e6bc2 commit 8a1b30f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion torchrec/distributed/benchmark/benchmark_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,8 @@
BENCH_SHARDING_TYPES: List[ShardingType] = [
ShardingType.TABLE_WISE,
ShardingType.ROW_WISE,
ShardingType.COLUMN_WISE,
# ShardingType.COLUMN_WISE,
# TODO: CW with FXJIT takes long time while profiling, doesn't cause an issue with no profiling in automatic benchmark
]

BENCH_COMPILE_MODES: List[CompileMode] = [
Expand Down

0 comments on commit 8a1b30f

Please sign in to comment.