Commit 75e521a0 authored by Masaki Kozuki's avatar Masaki Kozuki
Browse files

use newer DeviceUtils header

parent 386923b5

megatron/fused_kernels/layer_norm_cuda_kernel.cu

+1 −1
Original line number Diff line number Diff line
@@ -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>