Add fast math calls to new extern_function_name_map for OpenCL.

Author: mcourteaux

src/CodeGen_OpenCL_Dev.cpp

@@ -97,6 +97,20 @@ class CodeGen_OpenCL_Dev : public CodeGen_GPU_Dev {
             alias("fast_inverse", "native_recip");
             alias("fast_inverse_sqrt", "native_rsqrt");
 #undef alias
+
+            extern_function_name_map["fast_sin_f32"] = "native_sin";
+            extern_function_name_map["fast_cos_f32"] = "native_cos";
+            extern_function_name_map["fast_tan_f32"] = "native_tan";
+            extern_function_name_map["fast_exp_f32"] = "native_exp";
+            extern_function_name_map["fast_log_f32"] = "native_log";
+            extern_function_name_map["fast_pow_f32"] = "native_powr";
+
+            extern_function_name_map["fast_sin_f16"] = "half_sin";
+            extern_function_name_map["fast_cos_f16"] = "half_cos";
+            extern_function_name_map["fast_tan_f16"] = "half_tan";
+            extern_function_name_map["fast_exp_f16"] = "half_exp";
+            extern_function_name_map["fast_log_f16"] = "half_log";
+            extern_function_name_map["fast_pow_f16"] = "half_powr";
         }
         void add_kernel(Stmt stmt,
                         const std::string &name,
@@ -1140,12 +1154,6 @@ void CodeGen_OpenCL_Dev::init_module() {
                << "inline bool is_nan_f32(float x) {return isnan(x); }\n"
                << "inline bool is_inf_f32(float x) {return isinf(x); }\n"
                << "inline bool is_finite_f32(float x) {return isfinite(x); }\n"
-               << "#define fast_sin_f32 native_sin \n"
-               << "#define fast_cos_f32 native_cos \n"
-               << "#define fast_tan_f32 native_tan \n"
-               << "#define fast_exp_f32 native_exp \n"
-               << "#define fast_log_f32 native_log \n"
-               << "#define fast_pow_f32 native_powr \n"
                << "#define fast_inverse_f32 native_recip \n"
                << "#define fast_inverse_sqrt_f32 native_rsqrt \n";