Auto Format

taichi-dev · Dec 14, 2021 · c89a1b3 · c89a1b3
1 parent 70f3b05
commit c89a1b3
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/taichi/llvm/llvm_context.cpp b/taichi/llvm/llvm_context.cpp
@@ -521,7 +521,8 @@ std::size_t TaichiLLVMContext::get_type_size(llvm::Type *type) {
   return get_data_layout().getTypeAllocSize(type);
 }
 
-std::size_t TaichiLLVMContext::get_struct_element_offset(llvm::StructType *type, int idx) {
+std::size_t TaichiLLVMContext::get_struct_element_offset(llvm::StructType *type,
+                                                         int idx) {
   return get_data_layout().getStructLayout(type)->getElementOffset(idx);
 }
 

diff --git a/taichi/python/export_lang.cpp b/taichi/python/export_lang.cpp
@@ -403,7 +403,8 @@ void export_lang(py::module &m) {
       .def("num_active_indices",
            [](SNode *snode) { return snode->num_active_indices; })
       .def_readonly("cell_size_bytes", &SNode::cell_size_bytes)
-      .def_readonly("offset_bytes_in_parent_cell", &SNode::offset_bytes_in_parent_cell)
+      .def_readonly("offset_bytes_in_parent_cell",
+                    &SNode::offset_bytes_in_parent_cell)
       .def("begin_shared_exp_placement", &SNode::begin_shared_exp_placement)
       .def("end_shared_exp_placement", &SNode::end_shared_exp_placement);
 

diff --git a/taichi/struct/struct_llvm.cpp b/taichi/struct/struct_llvm.cpp
@@ -59,7 +59,8 @@ void StructCompilerLLVM::generate_types(SNode &snode) {
 
   for (int i = 0; i < snode.ch.size(); i++) {
     if (!snode.ch[i]->is_bit_level) {
-      snode.ch[i]->offset_bytes_in_parent_cell = tlctx_->get_struct_element_offset(ch_type, i);
+      snode.ch[i]->offset_bytes_in_parent_cell =
+          tlctx_->get_struct_element_offset(ch_type, i);
     }
   }
 

diff --git a/tests/python/test_snode_layout_inspection.py b/tests/python/test_snode_layout_inspection.py
@@ -26,7 +26,9 @@ def test_primitives():
 
     assert n1.offset_bytes_in_parent_cell == 0
     assert n2.offset_bytes_in_parent_cell == 2 * 32
-    assert n3.offset_bytes_in_parent_cell in [2 * 32 + 12 * 32, 2 * 32 + 16 * 32]
+    assert n3.offset_bytes_in_parent_cell in [
+        2 * 32 + 12 * 32, 2 * 32 + 16 * 32
+    ]
 
     assert x.snode.offset_bytes_in_parent_cell == 0
     assert y.snode.offset_bytes_in_parent_cell == 0