kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[None] [chore] Update to cutlass 4.3 (#8637)

Browse Source 
Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com>
This commit is contained in:
Kaiyu Xie 2025-11-28 08:54:34 +08:00 committed by GitHub
parent 2f8bd6fb36
commit 85b4c92d60
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
3rdparty/CMakeLists.txt vendored
View File

@ -19,7 +19,7 @@ FetchContent_Declare(
FetchContent_Declare(
cutlass
GIT_REPOSITORY https://github.com/NVIDIA/cutlass
GIT_TAG v4.2.1 # f3fde58372d33e9a5650ba7b80fc48b3b49d40c8
GIT_TAG v4.3.0 # e67e63c331d6e4b729047c95cf6b92c8454cba89
GIT_SHALLOW TRUE
SOURCE_SUBDIR
dont-add-this-project-with-add-subdirectory)

8
cpp/tensorrt_llm/cutlass_extensions/include/cutlass_extensions/gemm/collective/builders/sm90_gmma_builder_mixed_input.inl
View File

@ -177,13 +177,13 @@ public:
static constexpr int PipelineStages = IsMixedInput
? (IsArrayOfPointersGemm
? detail::compute_stage_count_or_override_single_affine_transformed_input<Sm90ReducedSmemCapacityBytes,
RealElementA, RealElementB, ElementScale, ElementZero, TileShape_MNK, StageCountType::bytes,
SmemAlignment>(StageCountType{})
RealElementA, RealElementB, ElementScale, ElementZero, TileShape_MNK, SmemAlignment,
StageCountType::bytes>(StageCountType{})
: detail::compute_stage_count_or_override_single_affine_transformed_input<
detail::sm90_smem_capacity_bytes, RealElementA, RealElementB, ElementScale, ElementZero,
TileShape_MNK, StageCountType::bytes, SmemAlignment>(StageCountType{}))
TileShape_MNK, SmemAlignment, StageCountType::bytes>(StageCountType{}))
: detail::compute_stage_count_or_override<detail::sm90_smem_capacity_bytes, ElementAMma, ElementBMma,
TileShape_MNK, StageCountType::bytes, SmemAlignment>(StageCountType{});
TileShape_MNK, SmemAlignment, StageCountType::bytes>(StageCountType{});
using DispatchPolicy = cute::conditional_t<IsMixedInput,
cute::conditional_t<IsArrayOfPointersGemm,