[ROCm][Inductor][CK] Fix codegen after ck signature change (pytorch#134483)

MakeArgument signature was changed in ROCm/composable_kernel#1453 adding splitK argument to universal gemm templates which are used to codegen addmm and matmul

(part of the series started at pytorch#125453 )

# Testing
`pytest test/inductor/test_ck_backend.py`

Pull Request resolved: pytorch#134483
Approved by: https://github.com/ColinPeppler

Author: tenpercent

torch/_inductor/codegen/rocm/ck_universal_gemm_template.py

@@ -57,6 +57,7 @@ class CKGemmTemplate(CKTemplate):
             LDB,
             std::array<ck::index_t, {{1 if has_bias else 0}}>{ {{'LDD' if has_bias else ''}} },
             LDC,
+            1, // kBatch
             PassThrough {}, // a_elementwise_op
             PassThrough {}, // b_elementwise_op
             {{epilogue}} // c_elementwise_op