fix cuda

HigherOrderCO · May 30, 2024 · 934d4a0 · 934d4a0
1 parent a51bcf2
commit 934d4a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/hvm.cu b/src/hvm.cu
@@ -261,7 +261,7 @@ __global__ void print_heatmap(GNet* gnet, u32 turn);
 // Utils
 // -----
 
-inline f32 clamp(f32 x, f32 min, f32 max) {
+__device__ __host__ f32 clamp(f32 x, f32 min, f32 max) {
   const f32 t = x < min ? min : x;
   return (t > max) ? max : t;
 }
@@ -561,7 +561,7 @@ __device__ __host__ inline bool is_cast(Numb word) {
 //                saturating if out of range and 0 if NaN
 // - i24  -> f24,
 //   u24  -> f24: casts to the "closest" float representing this integer.
-static inline Numb cast(Numb a, Numb b) {
+__device__ __host__ inline Numb cast(Numb a, Numb b) {
   if (get_sym(a) == TY_U24 && get_typ(b) == TY_U24) return b;
   if (get_sym(a) == TY_U24 && get_typ(b) == TY_I24) {
     // reinterpret bits