Skip to content

Commit cfbeabf

Browse files
committed
rename attributes to prevent duplicate naming
1 parent f9f2f8d commit cfbeabf

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

benchmarks/benchmark_e2e_fp8_sparse_linear.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,12 @@ def benchmark(num_tokens, hidden_size=8192, intermediate_size=8192):
4545
torch.ops.torchao.sparse24_sm90_sparsify(
4646
input_tensor,
4747
"cutlass",
48-
"srelu",
48+
"identity",
4949
"largest",
5050
dtype=torch.float8_e4m3fn,
5151
)
5252
)
53-
cusparse_time = benchmark_microseconds(torch._cslt_compress, input_tensor)
53+
cusparselt_time = benchmark_microseconds(torch._cslt_compress, input_tensor)
5454

5555
# bf16
5656
ffn_clone = (
@@ -130,9 +130,9 @@ def benchmark(num_tokens, hidden_size=8192, intermediate_size=8192):
130130
"fp8_c_sparse_time (us)": fp8_c_sparse_time,
131131
"fp8_c_activation_sparse_time (us)": fp8_c_activation_sparse_time,
132132
"ao_fast_sparsification_time (us)": ao_fast_sparsification_time,
133-
"cusparse*_compress_time (us)": cusparse_time,
133+
"cusparselt_compress_time (us)": cusparselt_time,
134134
"speedup": fp8_c_time / fp8_c_activation_sparse_time,
135-
"sparsify_speedup": cusparse_time / ao_fast_sparsification_time,
135+
"sparsify_speedup": cusparselt_time / ao_fast_sparsification_time,
136136
}
137137

138138

0 commit comments

Comments
 (0)