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:
@@ -109,6 +109,7 @@ struct Wmma<
|
||||
FragmentB const &B,
|
||||
FragmentC const &C) const {
|
||||
nvcuda::wmma::mma_sync(D, A, B, C);
|
||||
|
||||
}
|
||||
|
||||
#else
|
||||
@@ -186,7 +187,6 @@ struct Wmma<
|
||||
FragmentA const &A,
|
||||
FragmentB const &B,
|
||||
FragmentC const &C) const {
|
||||
|
||||
nvcuda::wmma::bmma_sync(D, A, B, C, nvcuda::wmma::experimental::bmmaBitOpXOR,
|
||||
nvcuda::wmma::experimental::bmmaAccumulateOpPOPC);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user