cleanup missing THCDeviceUtils.cuh header (#1177)

2a559c51 · X Wang · GitHub · 70d4a0ba · 2a559c51
Unverified Commit 2a559c51 authored Sep 28, 2021 by X Wang Committed by GitHub Sep 28, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

apex/contrib/csrc/multihead_attn/layer_norm.h apex/contrib/csrc/multihead_attn/layer_norm.h +1 -1

No files found.
--- a/apex/contrib/csrc/multihead_attn/layer_norm.h
+++ b/apex/contrib/csrc/multihead_attn/layer_norm.h
 #include "ATen/ATen.h"
-#include <THC/THCDeviceUtils.cuh>
+#include "ATen/cuda/DeviceUtils.cuh"

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