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

Hide streams and typinfo from nvrtc (#853)

Browse Source 
* Hide streams and typinfo from nvrtc

* Use __CUDACC_RTC__ instead CUDA_ARCH for guard
This commit is contained in:
Stepan Tezyunichev 2023-03-10 07:24:47 +03:00 committed by GitHub
parent 7e370c9637
commit 29801e348a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h
View File

@ -43,10 +43,11 @@
#include "cutlass/conv/conv3d_problem_size.h" #include "cutlass/conv/conv3d_problem_size.h"
#include "cutlass/gemm/threadblock/index_remat.h" #include "cutlass/gemm/threadblock/index_remat.h"
#if !defined(__CUDACC_RTC__)
#include <iostream> #include <iostream>
#include "cutlass/core_io.h" #include "cutlass/core_io.h"
#include "cutlass/trace.h" #include "cutlass/trace.h"
#endif