Merge pull request #128 from iotamudelta/master

Small fixes
ROCm · Aug 16, 2018 · d74ee71 · d74ee71
2 parents 7ccb5ad + a77bb25
commit d74ee71
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/aten/src/THC/THCNumerics.cuh b/aten/src/THC/THCNumerics.cuh
@@ -663,7 +663,7 @@ struct THCNumerics<double> {
   static inline __host__ __device__  double tanh (double a) { return  ::tanh(a); }
   static inline __host__ __device__  double erf  (double a) { return   ::erf(a); }
   static inline __host__ __device__  double erfc (double a) { return  ::erfc(a); }
-  static inline __host__ __device__  double abs  (double a) { return   ::abs(a); }
+  static inline __host__ __device__  double abs  (double a) { return   fabs(a); }
   static inline __host__ __device__  double round(double a) { return ::round(a); }
   static inline __host__ __device__  double frac (double a) { return a - ::trunc(a); }
   static inline __host__ __device__  double cinv (double a) { return 1.0 / a; }

diff --git a/tools/amd_build/disabled_features.yaml b/tools/amd_build/disabled_features.yaml
@@ -50,12 +50,6 @@
           "#include <nvfunctional>": ""
         }
       },
-      {
-        "path": "aten/src/ATen/native/cuda/Distributions.cu",
-        "s_constants": {
-          "#include <nvfunctional>": ""
-        }
-      },
       {
         "path": "aten/src/THC/THCNumerics.cuh",
         "s_constants": {