fix: force float for gaussian blur

clEsperanto · Jul 25, 2023 · 88ce412 · 88ce412
1 parent 70bc52f
commit 88ce412
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/clic/src/tier1.cpp b/clic/src/tier1.cpp
@@ -606,11 +606,17 @@ gaussian_blur_func(const Device::Pointer & device,
                    float                   sigma_y,
                    float                   sigma_z) -> Array::Pointer
 {
-  tier0::create_like(src, dst);
+  tier0::create_like(src, dst, dType::FLOAT);
+  Array::Pointer temp = src;
+  if (temp->dtype() != dType::FLOAT)
+  {
+    temp = Array::create(dst);
+    tier1::copy_func(device, src, temp);
+  }
   const KernelInfo kernel = { "gaussian_blur_separable", kernel::gaussian_blur_separable };
   tier0::execute_separable_func(device,
                                 kernel,
-                                src,
+                                temp,
                                 dst,
                                 { sigma_x, sigma_y, sigma_z },
                                 { sigma2kernelsize(sigma_x), sigma2kernelsize(sigma_y), sigma2kernelsize(sigma_z) });