.cu files should not include torch/extension.h

lostmsu · lostmsu · commit 605f78b08bfe · 2022-08-13T20:27:50.000-07:00
see facebookresearch/pytorch3d#1127 (comment)
diff --git a/csrc/megatron/fused_weight_gradient_dense_16bit_prec_cuda.cu b/csrc/megatron/fused_weight_gradient_dense_16bit_prec_cuda.cu
@@ -3,7 +3,6 @@
 #include <cstdlib>
 #include <cstring>
 
-#include <torch/extension.h>
 #include <ATen/ATen.h>
 #include <ATen/cuda/CUDAContext.h>
 
diff --git a/csrc/megatron/fused_weight_gradient_dense_cuda.cu b/csrc/megatron/fused_weight_gradient_dense_cuda.cu
@@ -3,7 +3,6 @@
 #include <cstdlib>
 #include <cstring>
 
-#include <torch/extension.h>
 #include <ATen/ATen.h>
 #include <ATen/cuda/CUDAContext.h>
 
diff --git a/csrc/megatron/generic_scaled_masked_softmax_cuda.cu b/csrc/megatron/generic_scaled_masked_softmax_cuda.cu
@@ -20,7 +20,6 @@
 #include <cuda_fp16.h>
 #include <cuda_profiler_api.h>
 #include <ATen/cuda/CUDAContext.h>
-#include <torch/extension.h>
 #include "generic_scaled_masked_softmax.h"
 #include "type_shim.h"
 
diff --git a/csrc/megatron/scaled_masked_softmax_cuda.cu b/csrc/megatron/scaled_masked_softmax_cuda.cu
@@ -20,7 +20,6 @@
 #include <cuda_fp16.h>
 #include <cuda_profiler_api.h>
 #include <ATen/cuda/CUDAContext.h>
-#include <torch/extension.h>
 #include "scaled_masked_softmax.h"
 #include "type_shim.h"
 
diff --git a/csrc/megatron/scaled_upper_triang_masked_softmax_cuda.cu b/csrc/megatron/scaled_upper_triang_masked_softmax_cuda.cu
@@ -20,7 +20,6 @@
 #include <cuda_fp16.h>
 #include <cuda_profiler_api.h>
 #include <ATen/cuda/CUDAContext.h>
-#include <torch/extension.h>
 #include "scaled_upper_triang_masked_softmax.h"
 #include "type_shim.h"
 
