fix contains

ROCm · Jan 7, 2025 · 7946afa · 7946afa
1 parent 03b843d
commit 7946afa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/gpu/jit.cpp b/test/gpu/jit.cpp
@@ -386,7 +386,7 @@ TEST_CASE(compile_math)
         if(contains({migraphx::shape::bool_type, migraphx::shape::tuple_type}, t))
             continue;
         auto name = migraphx::shape::cpp_type(t);
-        if((t == migraphx::shape::half_type) or (t == migraphx::shape::bf16_type))
+        if(contains({migraphx::shape::half_type, migraphx::shape::bf16_type}, t))
             name.insert(0, "migraphx::");
         data_types.push_back(name);
         // fp8 doesn't have vectorization support yet, therefore skip it for now.