Updates to fused epilogue (#383)

* Enhancements and fixes to fused GEMM and Convolution epilogue.
* Need to explicitly list cudart as unit test library dependency.
This commit is contained in:
Andrew Kerr
2021-12-17 16:04:43 -05:00
committed by GitHub
parent 4e666e1dfd
commit ec4f7e5194
24 changed files with 372 additions and 193 deletions

View File

@@ -296,7 +296,7 @@ int main() {
return -1;
}
if (!((props.major * 10 + props.minor) >= 80)) {
if (props.major * 10 + props.minor < 80) {
std::cerr << "Ampere Tensor Core operations must be run on a machine with compute capability at least 80."
<< std::endl;
notSupported = true;

View File

@@ -233,7 +233,7 @@ int run() {
tensor_b.device_ref(),
tensor_c_bias.device_ref(),
tensor_ref_d.device_ref(),
alpha, 0
alpha, ElementComputeEpilogue(0)
);
// Wait for kernels to finish

View File

@@ -45,6 +45,7 @@ function(cutlass_example_add_executable NAME)
PRIVATE
CUTLASS
cutlass_tools_util_includes
$<$<BOOL:${CUTLASS_ENABLE_CUBLAS}>:nvidia::cublas>
)
target_include_directories(