Skip to content

Commit 598a553

Browse files
EnragedAntelopejainapurva
authored andcommitted
Update base.h unit to unsigned int (#962)
1 parent 5b2439d commit 598a553

File tree

1 file changed

+2
-2
lines changed
  • torchao/csrc/cuda/sparse_marlin

1 file changed

+2
-2
lines changed

torchao/csrc/cuda/sparse_marlin/base.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ using I4 = Vec<int, 4>;
4444
// https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#matrix-fragments-for-mma-m16n8k16-with-floating-point-type
4545
using FragA = Vec<half2, 4>;
4646
using FragB = Vec<half2, 2>;
47-
using FragM = Vec<uint, 1>;
47+
using FragM = Vec<unsigned int, 1>;
4848
using FragC = Vec<float, 4>;
4949
using FragS = Vec<half2, 1>; // quantization scales
5050

51-
} // namespace torchao
51+
} // namespace torchao

0 commit comments

Comments
 (0)