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:
@@ -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;
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user