Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "82441460eff7c9ac1bd9ff63d406c58a9a190319"
Commit 5c79a278 authored by Masaki Kozuki's avatar Masaki Kozuki Committed by Hubert Lu
Browse files

THCDeviceUtils.cuh -> ATen/cuda/DeviceUtils.cuh (#1173)

parent 9f899769
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/layer_norm_cuda_kernel.cu
View file @ 5c79a278
#include "ATen/ATen.h" #include "ATen/ATen.h"
#include "ATen/AccumulateType.h" #include "ATen/AccumulateType.h"
#include "ATen/cuda/CUDAContext.h" #include "ATen/cuda/CUDAContext.h"
#include <THC/THCDeviceUtils.cuh> #include "ATen/cuda/DeviceUtils.cuh"
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment