leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 24 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/transform/thread/unaryOp.h
+8 -8
View File
@@ -54,19 +54,19 @@ class UnaryOp
static FragmentOut execute(FragmentIn &in)
{
static_assert(FragmentIn::kElements == FragmentOut::kElements, "Number of elements must match.");
static_assert(std::is_same<Transform, UnaryTransform::Identity>::value ||
std::is_same<Transform, UnaryTransform::Conjugate>::value,
static_assert(platform::is_same<Transform, UnaryTransform::Identity>::value ||
platform::is_same<Transform, UnaryTransform::Conjugate>::value,
"Unary Operator not supported.");
FragmentOut out;
if( std::is_same<Transform, UnaryTransform::Identity>::value )
if( platform::is_same<Transform, UnaryTransform::Identity>::value )
{
CUTLASS_PRAGMA_UNROLL
for(int i=0; i < FragmentIn::kElements; ++i){
out[i] = static_cast<typename FragmentOut::Element>(in[i]);
}
}
else if( std::is_same<Transform, UnaryTransform::Conjugate>::value )
else if( platform::is_same<Transform, UnaryTransform::Conjugate>::value )
{
for(int i=0; i < FragmentIn::kElements; ++i){
out[i] = conj(static_cast<typename FragmentOut::Element>(in[i]));
@@ -83,15 +83,15 @@ class UnaryOp<FragmentIn, FragmentIn, Transform>
CUTLASS_DEVICE
static FragmentIn execute(FragmentIn &in)
{
static_assert(std::is_same<Transform, UnaryTransform::Identity>::value ||
std::is_same<Transform, UnaryTransform::Conjugate>::value,
static_assert(platform::is_same<Transform, UnaryTransform::Identity>::value ||
platform::is_same<Transform, UnaryTransform::Conjugate>::value,
"Unary Operator not supported.");
if( std::is_same<Transform, UnaryTransform::Identity>::value )
if( platform::is_same<Transform, UnaryTransform::Identity>::value )
{
return in;
}
else if( std::is_same<Transform, UnaryTransform::Conjugate>::value )
else if( platform::is_same<Transform, UnaryTransform::Conjugate>::value )
{
for(int i=0; i < FragmentIn::kElements; ++i){
in[i] = conj(in[i]);