Skip to content

Commit 94e2e05

Browse files
Remove fix not needed anymore after moving CUTLASS pin to v3.9.0 (#2160)
1 parent 4850998 commit 94e2e05

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

torchao/dtypes/floatx/cutlass_semi_sparse_layout.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -155,12 +155,6 @@ def from_plain(
155155
):
156156
assert zero_point is None or torch.all(zero_point == 0)
157157

158-
# FIXME: remove this when CUTLASS PR #2110 merged.
159-
dtype = dense.dtype
160-
dense = dense.view(torch.uint8)
161-
dense[dense == 128] = 0
162-
dense = dense.view(dtype)
163-
164158
sparse, meta = to_sparse_semi_structured_cutlass_sm9x_f8(dense)
165159

166160
return cls(

0 commit comments

Comments
 (0)