CUTLASS 2.1 (#83)
CUTLASS 2.1 contributes: - BLAS-style host-side API added to CUTLASS Library - Planar Complex GEMM kernels targeting Volta and Turing Tensor Cores - Minor enhancements and bug fixes
This commit is contained in:
@@ -379,6 +379,7 @@ template <
|
||||
typename Element,
|
||||
typename Layout
|
||||
>
|
||||
CUTLASS_HOST_DEVICE
|
||||
bool TensorRef_aligned(TensorRef<Element, Layout> const &ref, int alignment) {
|
||||
|
||||
int const kStrideRank = Layout::kStrideRank;
|
||||
|
||||
Reference in New Issue
Block a user