leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Support for Group GEMM in CUTLASS Profiler for Geforce and Spark (#3092)

Browse Source 
Co-authored-by: dePaul Miller <23461061+depaulmillz@users.noreply.github.com>
This commit is contained in:
dePaul Miller
2026-03-06 17:36:29 -08:00
committed by GitHub
parent e5fcd125a5
commit 73c59c055c
8 changed files with 88 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/library/src/grouped_gemm_operation_3x.hpp
View File

@@ -563,9 +563,9 @@ public:
}
operator_args.mainloop.ptr_SFA =
static_cast<const typename Operator::GemmKernel::ElementSF**>(arguments->SFA);
static_cast<const typename CollectiveMainloop::ElementSF**>(arguments->SFA);
operator_args.mainloop.ptr_SFB =
static_cast<const typename Operator::GemmKernel::ElementSF**>(arguments->SFB);
static_cast<const typename CollectiveMainloop::ElementSF**>(arguments->SFB);
operator_args.mainloop.layout_SFA =
static_cast<typename CollectiveMainloop::InternalLayoutSFA*>(this->layout_SFA_device.data());