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

Add set_k_partition function (#624)

Browse Source 
A member function set_k_partition is required for the instatiation of cutlass::gemm::kernel::Gemm, even though SplitKSerial is false
This commit is contained in:
alexfreudenberg
2022-09-13 22:34:20 -04:00
committed by GitHub
parent 50ceed7154
commit 2cc2c7ba1f
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cutlass/epilogue/thread/conversion_op.h
+6
View File
@@ -92,6 +92,12 @@ public:
} }
/// Functionally required for serial reduction in the epilogue
CUTLASS_HOST_DEVICE
void set_k_partition(int k_partition, int k_partition_count) {
}
/// Returns true if source is needed based on state of runtime arguments /// Returns true if source is needed based on state of runtime arguments
CUTLASS_HOST_DEVICE CUTLASS_HOST_DEVICE
constexpr bool is_source_needed() const { constexpr bool is_source_needed() const {