Skip to content

GitLab

"git@developer.sourcefind.cn:one/TransferBench.git" did not exist on "a576c7612d85d24780d26f382a046ab45d2b1bf7"
Unverified Commit 2a559c51 authored by X Wang's avatar X Wang Committed by GitHub
Browse files

cleanup missing THCDeviceUtils.cuh header (#1177)

parent 70d4a0ba
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/contrib/csrc/multihead_attn/layer_norm.h
View file @ 2a559c51
#include "ATen/ATen.h" #include "ATen/ATen.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