fix matx util bug

nv-morpheus · Jan 31, 2024 · 41ea18b · 41ea18b
1 parent 8414ad8
commit 41ea18b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/morpheus/_lib/src/utilities/matx_util.cu b/morpheus/_lib/src/utilities/matx_util.cu
@@ -272,7 +272,7 @@ struct MatxUtil__MatxThreshold
         auto output_tensor = matx::make_tensor<bool>(static_cast<bool*>(output_data), output_shape);
 
         // Convert max value to bool
-        (output_tensor = matx::rmax(input_tensor) > (InputT)threshold).run(stream.value());
+        (output_tensor = matx::rmax(input_tensor, {1}) > (InputT)threshold).run(stream.value());
     }
 
     /**