Fix type

JuliaGPU · Aug 14, 2020 · 68244e4 · 68244e4
1 parent 8d68aeb
commit 68244e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/blas.jl b/src/blas.jl
@@ -67,7 +67,7 @@ function gemmEx!(transA::Char, transB::Char, alpha::Number, A, B, beta::Number,
                                 )
 
     GemmKernels.matmul(convert_matrix(A), convert_matrix(B), convert_matrix(C), convert_matrix(C), conf;
-                       transform_shared_to_regs_a = Transform.Elementwise(x -> multiply_fp16(x, alpha)),
+                       transform_shared_to_regs_a = Transform.Elementwise(x -> multiply_fp16(x, convert(Float16, alpha))),
                        transform_shared_to_regs_c = Transform.Elementwise(x -> x * beta))
 end