cleanup missing THCDeviceUtils.cuh header (NVIDIA#1177)

alealv · Sep 28, 2021 · 2a559c5 · 2a559c5
1 parent 70d4a0b
commit 2a559c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apex/contrib/csrc/multihead_attn/layer_norm.h b/apex/contrib/csrc/multihead_attn/layer_norm.h
@@ -1,5 +1,5 @@
 #include "ATen/ATen.h"
-#include <THC/THCDeviceUtils.cuh>
+#include "ATen/cuda/DeviceUtils.cuh"
 
 #include <cuda.h>
 #include <cuda_runtime.h>