We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 06003aa commit afa5a61Copy full SHA for afa5a61
benchmarks/bench_cutlass_fused_moe.py
@@ -29,7 +29,7 @@
29
FLOAT4_E2M1_MAX = 6.0
30
FLOAT8_E4M3_MAX = torch.finfo(torch.float8_e4m3fn).max
31
32
-num_ranks = 2
+num_ranks = 4
33
34
test_configs = [
35
# {
0 commit comments