added code to use the amdlibm librairy from amd when the theano flags set lib.amdlibm

theano/gof/utils.py

@@ -374,7 +374,7 @@ default_={
 'ProfileMode.n_apply_to_print':15,
 'ProfileMode.n_ops_to_print':20,
 'tensor_opt.local_elemwise_fusion':False,
-'scalar_basic.amdlibm':False,
+'lib.amdlibm':False,
 }
 
 

theano/scalar/basic.py

@@ -72,6 +72,23 @@ class Scalar(Type):
     def values_eq_approx(self, a, b, tolerance = 1e-4):
         return abs(a - b) / (a+b) < tolerance
 
+    def c_headers(self):
+        l=['<math.h>']
+        if utils.config.getboolean('lib.amdlibm'):
+            l+=['<amdlibm.h>']
+        return l
+
+    def c_libraries(self):
+        l=[]
+        if utils.config.getboolean('lib.amdlibm'):
+            l+=['amdlibm']
+        return l
+
+    def c_compile_args(self):
+        if utils.config.getboolean('lib.amdlibm'):
+            return ['-DREPLACE_WITH_AMDLIBM']
+        else: return []
+
     def __eq__(self, other):
         return type(self) == type(other) and other.dtype == self.dtype
 
@@ -220,7 +237,7 @@ class Scalar(Type):
 
     def c_code_cache_version(self):
         #return ()
-        return (3,)  #explicit T given in specialization of operator= lines.  This makes it compile with open64
+        return (4,)  #explicit T given in specialization of operator= lines.  This makes it compile with open64
         #2,
 
 

theano/tensor/basic.py

@@ -545,10 +545,13 @@ class TensorType(Type):
 
     def c_headers(self):
         """Override `CLinkerOp.c_headers` """
-        return []
+        return scal.Scalar(self.dtype).c_headers()
 
     def c_libraries(self):
-        return []
+        return scal.Scalar(self.dtype).c_libraries()
+
+    def c_compile_args(self):
+        return scal.Scalar(self.dtype).c_compile_args()
 
     def c_support_code(self):
         """Override `CLinkerOp.c_support_code` """