Skip to content

Commit 29801e3

Browse files
authored
Hide streams and typinfo from nvrtc (PaddlePaddle#853)
* Hide streams and typinfo from nvrtc * Use __CUDACC_RTC__ instead CUDA_ARCH for guard
1 parent 7e370c9 commit 29801e3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

include/cutlass/gemm/threadblock/threadblock_swizzle_streamk.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,11 @@
4343
#include "cutlass/conv/conv3d_problem_size.h"
4444
#include "cutlass/gemm/threadblock/index_remat.h"
4545

46+
#if !defined(__CUDACC_RTC__)
4647
#include <iostream>
4748
#include "cutlass/core_io.h"
4849
#include "cutlass/trace.h"
49-
50+
#endif
5051

5152

5253

0 commit comments

Comments
 (0)