Apply suggestions from code review

taichi-dev · Aug 25, 2022 · 5ed0e93 · 5ed0e93
1 parent 3451397
commit 5ed0e93
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/taichi/lang/impl.py b/python/taichi/lang/impl.py
@@ -39,7 +39,7 @@ def expr_init_local_tensor(shape, element_type, elements):
 
 @taichi_scope
 def make_matrix_expr(shape, element_type, elements):
-    return get_runtime().prog.current_ast_builder().make_local_matrix(
+    return get_runtime().prog.current_ast_builder().make_matrix_expr(
         shape, element_type, elements)
 
 

diff --git a/taichi/python/export_lang.cpp b/taichi/python/export_lang.cpp
@@ -291,7 +291,7 @@ void export_lang(py::module &m) {
       .def("insert_deactivate", &ASTBuilder::insert_snode_deactivate)
       .def("insert_activate", &ASTBuilder::insert_snode_activate)
       .def("insert_external_func_call", &ASTBuilder::insert_external_func_call)
-      .def("make_local_matrix", &ASTBuilder::make_matrix_expr)
+      .def("make_matrix_expr", &ASTBuilder::make_matrix_expr)
       .def("expr_alloca", &ASTBuilder::expr_alloca)
       .def("expr_alloca_local_tensor", &ASTBuilder::expr_alloca_local_tensor)
       .def("expr_alloca_shared_array", &ASTBuilder::expr_alloca_shared_array)