Formatting

src/targets/gpu/jit/ck_gemm_softmax_gemm.cpp

@@ -155,9 +155,8 @@ struct ck_gemm_softmax_gemm_compiler : compiler<ck_gemm_softmax_gemm_compiler>
 {
     static std::string get_layout(const shape& s)
     {
-        return s.strides().back() == 1
-                   ? "ck::tensor_layout::gemm::RowMajor"
-                   : "ck::tensor_layout::gemm::ColumnMajor";
+        return s.strides().back() == 1 ? "ck::tensor_layout::gemm::RowMajor"
+                                       : "ck::tensor_layout::gemm::ColumnMajor";
     }
 
     static std::string get_type(const shape& s)

tools/tune_models.py

@@ -40,9 +40,11 @@ def tune_models(models, batch_sizes, seq_len, n, existing):
     json_file = "ck_tuning_{}.json".format(time_stamp)
     for model in models:
         for batch in batch_sizes:
-            params = "--input-dim @sample {} 4 64 64 @timestep 1 @encoder_hidden_states {} 64 1024 --fp16 ".format(batch, batch)
+            params = "--input-dim @sample {} 4 64 64 @timestep 1 @encoder_hidden_states {} 64 1024 --fp16 ".format(
+                batch, batch)
             if "bert" in model:
-                params = "--fill1 input_ids --input-dim @input_ids {} {} ".format(batch, seq_len)
+                params = "--fill1 input_ids --input-dim @input_ids {} {} ".format(
+                    batch, seq_len)
             out = subprocess.run(
                 'MIGRAPHX_LOG_CK_GEMM=1 ../build/bin/driver run {} -g {} | grep \'ck_gemm.*: \[{{\' | sort -u >> {}'
                 .format(model, params, log_file),