Skip to content

Commit c5177a3

Browse files
committed
fix for new cutlass
1 parent eb87dac commit c5177a3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/mx_fp8_bf16.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,8 @@ void run_gemm(at::Tensor& a, at::Tensor& b, at::Tensor& a_scale,
9797
using StrideD = typename Gemm::GemmKernel::StrideD;
9898
using LayoutSFA = typename Gemm::GemmKernel::CollectiveMainloop::LayoutSFA;
9999
using LayoutSFB = typename Gemm::GemmKernel::CollectiveMainloop::LayoutSFB;
100-
using Sm100BlkScaledConfig = typename Gemm::GemmKernel::CollectiveMainloop::Sm100BlkScaledConfig;
100+
using Sm100BlkScaledConfig =
101+
typename Gemm::GemmKernel::CollectiveMainloop::Sm1xxBlkScaledConfig;
101102

102103
// Initialize strides using packed stride configuration
103104
auto stride_A = cutlass::make_cute_packed_stride(StrideA{}, make_shape(M, K, 1));

0 commit comments

Comments
 (0)