add passthrough for int32->float32 (#2062)

valarLip · web-flow · commit 2c563fecf76e · 2025-04-08T15:16:30.000-07:00
diff --git a/include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp b/include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp
@@ -357,6 +357,12 @@ struct PassThrough
         y = type_convert<half_t>(x);
     }
 
+    template <>
+    __host__ __device__ void operator()<float, int32_t>(float& y, const int32_t& x) const
+    {
+        y = type_convert<float>(x);
+    }
+
     template <>
     __host__ __device__ void operator()<bhalf_t, bhalf_t>(bhalf_t& y, const bhalf_t& x) const
     {

Original file line number	Diff line number	Diff line change
`@@ -357,6 +357,12 @@ struct PassThrough`
`357`	`357`	`y = type_convert<half_t>(x);`
`358`	`358`	`}`
`359`	`359`
	`360`	`+ template <>`
	`361`	`+ __host__ __device__ void operator()<float, int32_t>(float& y, const int32_t& x) const`
	`362`	`+ {`
	`363`	`+ y = type_convert<float>(x);`
	`364`	`+ }`
	`365`	`+`
`360`	`366`	`template <>`
`361`	`367`	`__host__ __device__ void operator()<bhalf_t, bhalf_t>(bhalf_t& y, const bhalf_t& x) const`
`362`	`368`	`{`