Putting back header inclusion (seems unused but still)

This commit is contained in:
Nicolas Patry 2023-07-20 15:46:51 +00:00
parent 5ca0508d02
commit f555dabca8
1 changed files with 1 additions and 1 deletions

View File

@ -3,7 +3,7 @@
#include <torch/extension.h> #include <torch/extension.h>
#include <c10/cuda/CUDAGuard.h> #include <c10/cuda/CUDAGuard.h>
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
// #include <cuda_runtime.h> #include <cuda_runtime.h>
#include <cuda_fp16.h> #include <cuda_fp16.h>
#include <cstdint> #include <cstdint>
#include <cstdio> #include <cstdio>