[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

python/taichi/lang/ast/ast_transformer.py

@@ -14,8 +14,8 @@
 from taichi.lang.ast.symbol_resolver import ASTResolver
 from taichi.lang.exception import TaichiSyntaxError, TaichiTypeError
 from taichi.lang.field import Field
-from taichi.lang.matrix import (Matrix, MatrixType, _PyScopeMatrixImpl,
-                                _TiScopeMatrixImpl, Vector)
+from taichi.lang.matrix import (Matrix, MatrixType, Vector, _PyScopeMatrixImpl,
+                                _TiScopeMatrixImpl)
 from taichi.lang.snode import append
 from taichi.lang.util import in_taichi_scope, is_taichi_class, to_taichi_type
 from taichi.types import (annotations, ndarray_type, primitive_types,
@@ -488,7 +488,8 @@ def build_Call(ctx, node):
             node.ptr = impl.ti_format(*args, **keywords)
             return node.ptr
 
-        if isinstance(node.func, ast.Attribute) and func == Matrix or func == Vector:
+        if isinstance(node.func,
+                      ast.Attribute) and func == Matrix or func == Vector:
             node.ptr = matrix.make_matrix(*args, **keywords)
             return node.ptr
 

python/taichi/lang/matrix.py

@@ -103,12 +103,14 @@ def make_matrix(arr, dt=None, suppress_warning=False, is_ref=False, **kwargs):
     cast = (lambda x: ops_mod.cast(x, dt)) if dt else (
         lambda x: x if isinstance(x, expr.Expr) else expr.Expr(x))
     if len(arr) == 0:
-        return  impl.expr_init(impl.expr_init_matrix([0], dt, []))
+        return impl.expr_init(impl.expr_init_matrix([0], dt, []))
     if not isinstance(arr[0], Iterable):
-        return impl.expr_init(impl.expr_init_matrix([len(arr)], dt,
-                                     [cast(elt).ptr for elt in arr]))
-    return impl.expr_init(impl.expr_init_matrix([len(arr), len(arr[0])], dt,
-                                 [cast(elt).ptr for row in arr for elt in row]))
+        return impl.expr_init(
+            impl.expr_init_matrix([len(arr)], dt,
+                                  [cast(elt).ptr for elt in arr]))
+    return impl.expr_init(
+        impl.expr_init_matrix([len(arr), len(arr[0])], dt,
+                              [cast(elt).ptr for row in arr for elt in row]))
 
 
 class _MatrixBaseImpl:

taichi/codegen/llvm/codegen_llvm.cpp

@@ -1484,8 +1484,7 @@ void TaskCodeGenLLVM::visit(GlobalStoreStmt *stmt) {
       TI_NOT_IMPLEMENTED;
     }
   } else {
-    TI_TRACE("Store {} to {}", stmt->val->name(),
-             stmt->dest->name());
+    TI_TRACE("Store {} to {}", stmt->val->name(), stmt->dest->name());
     llvm_val[stmt->val]->dump();
     llvm_val[stmt->dest]->dump();
     builder->CreateStore(llvm_val[stmt->val], llvm_val[stmt->dest]);

taichi/ir/statements.h

@@ -367,9 +367,11 @@ class PtrOffsetStmt : public Stmt {
 
   bool is_local_ptr() const {
     if (origin->is<AllocaStmt>() || origin->is<GlobalTemporaryStmt>()) {
-      auto is_tensor_type = origin->ret_type->is<PointerType>() ?
-                          origin->ret_type->cast<PointerType>()->get_pointee_type()->is<TensorType>() :
-                          origin->ret_type->is<TensorType>();
+      auto is_tensor_type = origin->ret_type->is<PointerType>()
+                                ? origin->ret_type->cast<PointerType>()
+                                      ->get_pointee_type()
+                                      ->is<TensorType>()
+                                : origin->ret_type->is<TensorType>();
       TI_ASSERT_INFO(is_tensor_type,
                      "PtrOffsetStmt can only be used for Alloca (TensorType).");
     }