Skip to content

Commit 85b4c92

Browse files
authored
[None] [chore] Update to cutlass 4.3 (#8637)
Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com>
1 parent 2f8bd6f commit 85b4c92

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

3rdparty/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ FetchContent_Declare(
1919
FetchContent_Declare(
2020
cutlass
2121
GIT_REPOSITORY https://github.com/NVIDIA/cutlass
22-
GIT_TAG v4.2.1 # f3fde58372d33e9a5650ba7b80fc48b3b49d40c8
22+
GIT_TAG v4.3.0 # e67e63c331d6e4b729047c95cf6b92c8454cba89
2323
GIT_SHALLOW TRUE
2424
SOURCE_SUBDIR
2525
dont-add-this-project-with-add-subdirectory)

cpp/tensorrt_llm/cutlass_extensions/include/cutlass_extensions/gemm/collective/builders/sm90_gmma_builder_mixed_input.inl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -177,13 +177,13 @@ public:
177177
static constexpr int PipelineStages = IsMixedInput
178178
? (IsArrayOfPointersGemm
179179
? detail::compute_stage_count_or_override_single_affine_transformed_input<Sm90ReducedSmemCapacityBytes,
180-
RealElementA, RealElementB, ElementScale, ElementZero, TileShape_MNK, StageCountType::bytes,
181-
SmemAlignment>(StageCountType{})
180+
RealElementA, RealElementB, ElementScale, ElementZero, TileShape_MNK, SmemAlignment,
181+
StageCountType::bytes>(StageCountType{})
182182
: detail::compute_stage_count_or_override_single_affine_transformed_input<
183183
detail::sm90_smem_capacity_bytes, RealElementA, RealElementB, ElementScale, ElementZero,
184-
TileShape_MNK, StageCountType::bytes, SmemAlignment>(StageCountType{}))
184+
TileShape_MNK, SmemAlignment, StageCountType::bytes>(StageCountType{}))
185185
: detail::compute_stage_count_or_override<detail::sm90_smem_capacity_bytes, ElementAMma, ElementBMma,
186-
TileShape_MNK, StageCountType::bytes, SmemAlignment>(StageCountType{});
186+
TileShape_MNK, SmemAlignment, StageCountType::bytes>(StageCountType{});
187187

188188
using DispatchPolicy = cute::conditional_t<IsMixedInput,
189189
cute::conditional_t<IsArrayOfPointersGemm,

0 commit comments

Comments
 (0)