format

NVIDIA · Jan 29, 2025 · 2fe5579 · 2fe5579
1 parent 4e5ab6e
commit 2fe5579
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/csrc/device_lower/pass/index.cpp b/csrc/device_lower/pass/index.cpp
@@ -2164,7 +2164,8 @@ void IndexLowering::handle(const LoadStoreOp* ldst) {
         auto index = IrBuilder::create<Val>(
             std::vector<int64_t>{0, 0},
             ArrayType{std::make_shared<DataType>(DataType::UInt16), 2});
-        in = IrBuilder::create<kir::TensorIndex>(tv, index, DataType::TMemAddress);
+        in = IrBuilder::create<kir::TensorIndex>(
+            tv, index, DataType::TMemAddress);
       } else {
         in = lowerSrcIndex(
             ldst->in(),
@@ -2179,7 +2180,8 @@ void IndexLowering::handle(const LoadStoreOp* ldst) {
         auto index = IrBuilder::create<Val>(
             std::vector<int64_t>{0, 0},
             ArrayType{std::make_shared<DataType>(DataType::UInt16), 2});
-        out = IrBuilder::create<kir::TensorIndex>(tv, index, DataType::TMemAddress);
+        out = IrBuilder::create<kir::TensorIndex>(
+            tv, index, DataType::TMemAddress);
       } else {
         out = lowerDstIndex(
             ldst->out(), {}, ir_utils::isCpAsyncOp(ldst), as_type);