more explicit template instantiations

flexflow · Nov 3, 2022 · 0ad6d3a · 0ad6d3a
1 parent 0bf5e99
commit 0ad6d3a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/runtime/hip_helper.cpp b/src/runtime/hip_helper.cpp
@@ -360,6 +360,10 @@ template __global__ void
     add_kernel<float>(float *dst, float const *src, size_t size);
 template __global__ void
     add_kernel<double>(double *dst, double const *src, size_t size);
+template __global__ void
+    add_kernel<int>(int *dst, int const *src, size_t size);
+template __global__ void
+    add_kernel<long>(long *dst, long const *src, size_t size);
 
 template __global__ void
     copy_kernel<float>(float *dst, float const *src, coord_t size);