Skip to content

GitLab

Unverified Commit 251cddaf authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

Merge pull request #15 from syed-ahmed/master 

Adding changes due to https://github.com/pytorch/pytorch/pull/8660
parents 06d035e3 bddbbdcb
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
csrc/scale_cuda.cu
View file @ 251cddaf
#include <ATen/ATen.h> #include <ATen/ATen.h>
// #include "ATen/AccumulateType.h" // #include "ATen/AccumulateType.h"
#include "ATen/cuda/CUDATensorMethods.cuh" #include "ATen/cuda/CUDATensorMethods.cuh"
#include "ATen/cuda/CUDATypeConversion.cuh" // #include "ATen/cuda/CUDATypeConversion.cuh"
// #include <THC/THCTensorMathReduce.cuh> // #include <THC/THCTensorMathReduce.cuh>
#include <THC/THCGeneral.h> #include <THC/THCGeneral.h>
......
csrc/weight_norm_bwd_cuda.cu
View file @ 251cddaf
...@@ -9,7 +9,7 @@ ...@@ -9,7 +9,7 @@
#endif #endif
#include "ATen/cuda/CUDATensorMethods.cuh" #include "ATen/cuda/CUDATensorMethods.cuh"
#include "ATen/cuda/CUDATypeConversion.cuh" // #include "ATen/cuda/CUDATypeConversion.cuh"
// #include <THC/THCTensorMathReduce.cuh> // #include <THC/THCTensorMathReduce.cuh>
template template
......
csrc/weight_norm_fwd_cuda.cu
View file @ 251cddaf
...@@ -9,7 +9,7 @@ ...@@ -9,7 +9,7 @@
#endif #endif
#include "ATen/cuda/CUDATensorMethods.cuh" #include "ATen/cuda/CUDATensorMethods.cuh"
#include "ATen/cuda/CUDATypeConversion.cuh" // #include "ATen/cuda/CUDATypeConversion.cuh"
// #include <THC/THCTensorMathReduce.cuh> // #include <THC/THCTensorMathReduce.cuh>
template template
......
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