Skip to content

Commit ada178c

Browse files
committed
[example] keep gemm_sp on hopper
1 parent 9d755bf commit ada178c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/gemm_sp/example_gemm_sp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ def gemm_sp_fp16(
9494
T.copy(A_sparse[by * block_M, k * block_K // 2], A_shared)
9595
T.copy(E[by * block_M, k * block_K // e_factor], E_shared)
9696
T.copy(B[k * block_K, bx * block_N], B_shared)
97-
T.gemm_sp_v2(A_shared, E_shared, B_shared, C_local, False, False, policy=policy)
97+
T.gemm_sp(A_shared, E_shared, B_shared, C_local, False, False, policy=policy)
9898

9999
T.copy(C_local, C_shared)
100100
T.copy(C_shared, C[by * block_M, bx * block_N])

0 commit comments

Comments
 (0)