From daddb71229ec42b4629a1822466929707c9879d5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleksandar=20Samard=C5=BEi=C4=87?= Date: Sat, 26 Apr 2025 11:08:46 +0000 Subject: [PATCH] Move CUTLASS pin to v3.9.0 --- third_party/cutlass | 2 +- torchao/csrc/cuda/mx_kernels/mx_fp_cutlass_kernels.cu | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/third_party/cutlass b/third_party/cutlass index afa1772203..e94e888df3 160000 --- a/third_party/cutlass +++ b/third_party/cutlass @@ -1 +1 @@ -Subproject commit afa1772203677c5118fcd82537a9c8fefbcc7008 +Subproject commit e94e888df3551224738bfa505787b515eae8352f diff --git a/torchao/csrc/cuda/mx_kernels/mx_fp_cutlass_kernels.cu b/torchao/csrc/cuda/mx_kernels/mx_fp_cutlass_kernels.cu index 87aa0b0501..97a243bb70 100644 --- a/torchao/csrc/cuda/mx_kernels/mx_fp_cutlass_kernels.cu +++ b/torchao/csrc/cuda/mx_kernels/mx_fp_cutlass_kernels.cu @@ -107,7 +107,7 @@ void run_gemm(at::Tensor& a, at::Tensor& b, at::Tensor& a_scale, using StrideD = typename Gemm::GemmKernel::StrideD; using LayoutSFA = typename Gemm::GemmKernel::CollectiveMainloop::LayoutSFA; using LayoutSFB = typename Gemm::GemmKernel::CollectiveMainloop::LayoutSFB; - using Sm100BlkScaledConfig = typename Gemm::GemmKernel::CollectiveMainloop::Sm100BlkScaledConfig; + using Sm100BlkScaledConfig = typename Gemm::GemmKernel::CollectiveMainloop::Sm1xxBlkScaledConfig; // Initialize strides using packed stride configuration auto stride_A = cutlass::make_cute_packed_stride(StrideA{}, make_shape(M, K, 1));