@@ -881,10 +881,9 @@ Stmt Copy::LowerBulkCopy(const LowerArgs &T, arith::Analyzer *analyzer,
881881    auto  stride = as_const_int (shared_layout->InputShape ()[0 ]);
882882    auto  continuous = as_const_int (shared_layout->InputShape ()[1 ]);
883883    ICHECK (stride != nullptr  && continuous != nullptr );
884-     if  (StructuralEqual ()(
885-                    shared_layout,
886-                    makeGemmABLayoutPadded (*stride, *continuous,
887-                                           shared_tensor->dtype .bits ()))) {
884+     if  (StructuralEqual ()(shared_layout, makeGemmABLayoutPadded (
885+                                              *stride, *continuous,
886+                                              shared_tensor->dtype .bits ()))) {
888887      desc.swizzle  = static_cast <int >(CU_TENSOR_MAP_SWIZZLE_NONE);
889888    } else  if  (StructuralEqual ()(
890889                   shared_layout,
@@ -932,14 +931,12 @@ Stmt Copy::LowerBulkCopy(const LowerArgs &T, arith::Analyzer *analyzer,
932931
933932  if  (desc.swizzle  == static_cast <int >(CU_TENSOR_MAP_SWIZZLE_NONE) &&
934933      inner_box_dim_ % 256  != 0 ) {
935-     LOG (WARNING) << " TMA bulk copy fallback to normal copy: " 
936-                  << " non-swizzled global layout with inner_box_dim_ % 256 != 0. " 
937-                  << " inner_box_dim_: " 
938-                  << " , src buffer: " name 
939-                  << " , dst buffer: " name 
940-                  << " , smem_box: " smem_box 
941-                  << " , src shape: " shape 
942-                  << " , dst shape: " shape ;
934+     LOG (WARNING)
935+         << " TMA bulk copy fallback to normal copy: " 
936+         << " non-swizzled global layout with inner_box_dim_ % 256 != 0. " 
937+         << " inner_box_dim_: " " , src buffer: " name 
938+         << " , dst buffer: " name  << " , smem_box: " smem_box 
939+         << " , src shape: " shape  << " , dst shape: " shape ;
943940    return  LowerNormalCopy (T, analyzer);
944941  }
945942  //  Check inner_box_dim_ for each swizzle type in a cleaner way
0 commit comments