[Bugfix] Pass pre commit check

Author: Lukinon

src/op/gemm.cc

@@ -865,12 +865,12 @@ LayoutMap GemmNode::InferLayout(const LayoutInferArgs &T,
         << "CDNA gemm (FMMA) only supports C in local.fragment scope, got "
         << c_.scope();
     if (TargetIsDCU(T.target)) {
-      auto fragment =
-          makeGemmFragmentCDCU(m_, n_, m_ / warp_m, n_ / warp_n, c_->dtype.bits());
+      auto fragment = makeGemmFragmentCDCU(m_, n_, m_ / warp_m, n_ / warp_n,
+                                           c_->dtype.bits());
       results.Set(c_, fragment->BindThreadRange(thread_range));
     } else {
-      auto fragment =
-          makeGemmFragmentCCDNA(m_, n_, m_ / warp_m, n_ / warp_n, c_->dtype.bits());
+      auto fragment = makeGemmFragmentCCDNA(m_, n_, m_ / warp_m, n_ / warp_n,
+                                            c_->dtype.bits());
       results.Set(c_, fragment->BindThreadRange(thread_range));
     }
     if (a_.scope() == "shared" || a_.scope() == "shared.dyn") {

tilelang/contrib/rocm.py

@@ -226,11 +226,10 @@ def have_matrixcore(compute_version=None):
     return False
 
 
-
 @tvm_ffi.register_global_func("tvm_callback_rocm_get_arch", override=True)
 def get_rocm_arch(rocm_path="/opt/rocm"):
-# @tvm.ffi.register_func("tvm_callback_rocm_get_arch", override=True)
-# def get_rocm_arch(rocm_path="/opt/dtk"):
+    # @tvm.ffi.register_func("tvm_callback_rocm_get_arch", override=True)
+    # def get_rocm_arch(rocm_path="/opt/dtk"):
     """Utility function to get the AMD GPU architecture
 
     Parameters