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

Address CR

Browse Source
This commit is contained in:
KeDengMS 2021-04-19 23:36:06 +00:00
parent 83036ed646
commit 0b74c8f473
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/cutlass/epilogue/thread/linear_combination_gelu.h
View File

@ -134,6 +134,7 @@ public:
/// Functionally required for serial reduction in the epilogue /// Functionally required for serial reduction in the epilogue
CUTLASS_HOST_DEVICE CUTLASS_HOST_DEVICE
void set_k_partition(int k_partition, int k_partition_count) { void set_k_partition(int k_partition, int k_partition_count) {
CUTLASS_UNUSED(k_partition_count);
if (k_partition) { if (k_partition) {
beta_ = ElementCompute(1); beta_ = ElementCompute(1);
} }