1 | #pragma once |
---|---|
2 | |
3 | #include <cuda.h> |
4 | #include <cuda_runtime.h> |
5 | #include <cuda_fp16.h> |
6 | |
7 | #include <c10/macros/Export.h> |
8 | |
9 | // Use TORCH_CUDA_CPP_API or TORCH_CUDA_CU_API for exports from this folder |
10 |
1 | #pragma once |
---|---|
2 | |
3 | #include <cuda.h> |
4 | #include <cuda_runtime.h> |
5 | #include <cuda_fp16.h> |
6 | |
7 | #include <c10/macros/Export.h> |
8 | |
9 | // Use TORCH_CUDA_CPP_API or TORCH_CUDA_CU_API for exports from this folder |
10 |