fix(codegen): add herf function definition

apache · Aug 27, 2020 · 505421e · 505421e
1 parent c958bc1
commit 505421e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/target/source/literal/cuda_half_t.h b/src/target/source/literal/cuda_half_t.h
@@ -308,6 +308,10 @@ static inline __device__ __host__ half htanh(half x) {
   float result = tanhf(tmp_x);
   return __float2half(result);
 }
+
+static inline __device__ __host__ half herf(half x) {
+  return __float2half(erf(__half2float(x)));
+}
 #endif
 )";