use newer DeviceUtils header

75e521a0 · Masaki Kozuki · 386923b5 · 75e521a0
Commit 75e521a0 authored Oct 06, 2021 by Masaki Kozuki
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

megatron/fused_kernels/layer_norm_cuda_kernel.cu megatron/fused_kernels/layer_norm_cuda_kernel.cu +1 -1

No files found.
--- a/megatron/fused_kernels/layer_norm_cuda_kernel.cu
+++ b/megatron/fused_kernels/layer_norm_cuda_kernel.cu
@@ -21,7 +21,7 @@
 #include "ATen/ATen.h"
 #include "ATen/AccumulateType.h"
 #include "ATen/cuda/CUDAContext.h"
-#include <THC/THCDeviceUtils.cuh>
+#include "ATen/cuda/DeviceUtils.cuh"

 #include <cuda.h>
 #include <cuda_runtime.h>