Formatting

src/onnx/parse_where.cpp

@@ -31,6 +31,8 @@ namespace migraphx {
 inline namespace MIGRAPHX_INLINE_NS {
 namespace onnx {
 
+MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_ENABLE_CK)
+
 struct parse_where : op_parser<parse_where>
 {
     std::vector<op_desc> operators() const { return {{"Where"}}; }
@@ -57,6 +59,12 @@ struct parse_where : op_parser<parse_where>
                 compute_broadcasted_lens(args[0]->get_shape().lens(), args[1]->get_shape().lens());
             lens = compute_broadcasted_lens(lens, args[2]->get_shape().lens());
 
+            if (enabled(MIGRAPHX_ENABLE_CK{}))
+            {
+                // Convert condition tensor to int32 to work around CK not supporting bool type
+                args[0] = info.add_instruction(make_op("convert", {{"target_type", shape::int32_type}}), args[0]);
+            }
+            
             if(args[0]->get_shape().lens() != lens)
             {
                 args[0] =

src/targets/gpu/jit/ck_gemm.cpp

@@ -355,7 +355,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
         const auto& c_shape = inputs.back();
         auto tuning_value   = v.get("tuning_value", 4);
         if(not v.contains("tuning_value"))
-            tuning_value = get_tuning_for({a_shape, b_shape, c_shape});
+            tuning_value = get_tuning_for({a_shape, b_shape, c_shape});   
         auto batch_count = get_batch_count(c_shape);
         auto problem     = create_problem(inputs, v);
 
@@ -418,7 +418,10 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
     {
         auto shapes       = to_shapes(ins->inputs());
         auto v            = create_settings(ins, op);
-        v["tuning_value"] = solution;
+        if (solution.is_null())
+            v["tuning_value"] = 4;
+        else
+            v["tuning_value"] = solution;
         return {compile_op(ctx, shapes, v),
                 [=](module& m, instruction_ref ins2, const operation& code_object) {
                     if(enabled(MIGRAPHX_LOG_CK_GEMM{}))