Skip to content

GitLab

"...text-generation-inference.git" did not exist on "64def8e298efc50caf23cc051d38fd3182fc242d"
Commit abb6e5ba authored by X Wang's avatar X Wang Committed by Hubert Lu
Browse files

cleanup missing THCDeviceUtils.cuh header (#1177)

parent 5c79a278
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/contrib/csrc/multihead_attn/layer_norm.h
View file @ abb6e5ba
#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