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