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