leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 26 Packages Projects Releases Wiki Activity

2.9 fixes for nvrtc (#480)

Browse Source 
* Use platform::is_same instead of std::is_same

* Don't hide cuComplex include from nvrtc

* Typo fixed

* Remove comment rename
This commit is contained in:
Stepan Tezyunichev
2022-04-29 09:06:52 -04:00
committed by GitHub
parent 21c1fa3849
commit 86ce09aed1
13 changed files with 36 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cutlass/gemm/device/rank_2k.h
+1 -1
View File
@@ -149,7 +149,7 @@ class Rank2K {
static int const kUpdateRank = 2;
// static asserts for rank 2k update kernel
static_assert(std::is_same<LayoutA, LayoutB>::value,
static_assert(platform::is_same<LayoutA, LayoutB>::value,
"Rank 2K update operator support same layouts for operandA and B");
/// Define the kernel
include/cutlass/gemm/device/symm.h
+1 -1
View File
@@ -153,7 +153,7 @@ class Symm {
static BlasMode const kBlasMode = BlasMode_;
// static asserts for symm update kernel
static_assert(std::is_same<LayoutA, LayoutB>::value,
static_assert(platform::is_same<LayoutA, LayoutB>::value,
"SYMM update operator support same layouts for operand A and B");
/// Define the kernel
include/cutlass/gemm/kernel/default_gemm.h
+3 -3
View File
@@ -209,7 +209,7 @@ struct DefaultGemm<ElementA, LayoutA, kAlignmentA, ElementB, LayoutB, kAlignment
2, ThreadblockShape, typename Mma::Operator, kPartitionsK, EpilogueOutputOp,
EpilogueOutputOp::kCount>::Epilogue;
using Epilogue = typename cutlass::platform::conditional<cutlass::platform::is_same<LayoutC, layout::RowMajor>::value,
using Epilogue = typename cutlass::platform::conditional<platform::is_same<LayoutC, layout::RowMajor>::value,
RegularEpilogue,
Affine2Epilogue>::type;
@@ -672,7 +672,7 @@ struct DefaultGemm<
kEpilogueElementsPerAccess
>::Epilogue;
using Epilogue = typename cutlass::platform::conditional<cutlass::platform::is_same<LayoutC, layout::RowMajor>::value,
using Epilogue = typename cutlass::platform::conditional<platform::is_same<LayoutC, layout::RowMajor>::value,
RegularEpilogue,
Affine2Epilogue>::type;
@@ -780,7 +780,7 @@ struct DefaultGemm<ElementA,
kEpilogueElementsPerAccess
>::Epilogue;
using Epilogue = typename cutlass::platform::conditional<cutlass::platform::is_same<LayoutC, layout::RowMajor>::value,
using Epilogue = typename cutlass::platform::conditional<platform::is_same<LayoutC, layout::RowMajor>::value,
RegularEpilogue,
Affine2Epilogue>::type;
include/cutlass/gemm/kernel/default_gemm_grouped.h
+2 -2
View File
@@ -183,7 +183,7 @@ struct DefaultGemmGrouped<
> {
// If true, we must construct a 'transposed-and-exchanged' Mma operator.
static bool const kInternalTranspose = std::is_same<LayoutC, layout::ColumnMajor>::value;
static bool const kInternalTranspose = platform::is_same<LayoutC, layout::ColumnMajor>::value;
using MapArguments = kernel::detail::MapArguments<
ElementA,
@@ -307,7 +307,7 @@ struct DefaultGemmGrouped<
> {
// If true, we must construct a 'transposed-and-exchanged' Mma operator.
static bool const kInternalTranspose = std::is_same<LayoutC, layout::ColumnMajor>::value;
static bool const kInternalTranspose = platform::is_same<LayoutC, layout::ColumnMajor>::value;
using MapArguments = kernel::detail::MapArguments<
ElementA,
include/cutlass/gemm/kernel/gemv.h
+1 -1
View File
@@ -67,7 +67,7 @@ public:
using LayoutA = layout::ColumnMajor;
using TensorRefA = TensorRef<ElementA, LayoutA>;
static_assert(std::is_same<LayoutA, LayoutA_>::value,
static_assert(platform::is_same<LayoutA, LayoutA_>::value,
"Only supported for column-major A matrix");
using ElementB = ElementB_;
include/cutlass/gemm/threadblock/default_mma.h
+2 -2
View File
@@ -632,8 +632,8 @@ struct DefaultMma<int8_t, LayoutA, kAlignmentA, int8_t, LayoutB, kAlignmentB,
using ElementB = int8_t;
using OperatorClass = arch::OpClassSimt;
static const bool transposeA = cutlass::platform::is_same< LayoutA, layout::ColumnMajor >::value;
static const bool transposeB = cutlass::platform::is_same< LayoutB, layout::RowMajor >::value;
static const bool transposeA = platform::is_same< LayoutA, layout::ColumnMajor >::value;
static const bool transposeB = platform::is_same< LayoutB, layout::RowMajor >::value;
// Define the MmaCore components
using MmaCore = typename cutlass::gemm::threadblock::DefaultMmaCore<