Updates to fused epilogue (#383)

* Enhancements and fixes to fused GEMM and Convolution epilogue. * Need to explicitly list cudart as unit test library dependency.
2021-12-17 16:04:43 -05:00
parent 4e666e1dfd
commit ec4f7e5194
24 changed files with 372 additions and 193 deletions
@@ -109,6 +109,12 @@ static char const* cutlassGetStatusString(cutlass::Status status) {

 ////////////////////////////////////////////////////////////////////////////////////////////////////

+
+#ifndef CUTLASS_CONV_UNIT_TEST_RIGOROUS_SIZE_ENABLED
+#define CUTLASS_CONV_UNIT_TEST_RIGOROUS_SIZE_ENABLED 0
+#endif
+
+
 // CUDA 10.1 introduces the mma instruction
 #if !defined(CUTLASS_ENABLE_TENSOR_CORE_MMA)
 #define CUTLASS_ENABLE_TENSOR_CORE_MMA 0