Rename cg into sparse_cg.

taichi-dev · Apr 27, 2023 · 3371aae · 3371aae
1 parent ead0549
commit 3371aae
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/python/taichi/linalg/__init__.py b/python/taichi/linalg/__init__.py
@@ -1,6 +1,6 @@
 """Taichi support module for sparse matrix operations.
 """
-from taichi.linalg.cg import CG
+from taichi.linalg.sparse_cg import SparseCG
 from taichi.linalg.sparse_matrix import *
 from taichi.linalg.sparse_solver import SparseSolver
 from taichi.linalg.matrixfree_cg import *
diff --git a/python/taichi/linalg/cg.py → python/taichi/linalg/sparse_cg.py b/python/taichi/linalg/cg.py → python/taichi/linalg/sparse_cg.py
@@ -6,7 +6,7 @@
 from taichi.types import f32, f64
 
 
-class CG:
+class SparseCG:
     def __init__(self, A, b, x0=None, max_iter=50, atol=1e-6):
         self.dtype = A.dtype
         self.ti_arch = get_runtime().prog.config().arch

diff --git a/tests/python/test_cg.py → tests/python/test_sparse_cg.py b/tests/python/test_cg.py → tests/python/test_sparse_cg.py
@@ -28,7 +28,7 @@ def fill(
 
     fill(Abuilder, A_psd, b)
     A = Abuilder.build(dtype=ti_dtype)
-    cg = ti.linalg.CG(A, b, x0, max_iter=50, atol=1e-6)
+    cg = ti.linalg.SparseCG(A, b, x0, max_iter=50, atol=1e-6)
     x, exit_code = cg.solve()
     res = np.linalg.solve(A_psd, b.to_numpy())
     assert exit_code == True
@@ -59,7 +59,7 @@ def fill(
 
     fill(Abuilder, A_psd, b)
     A = Abuilder.build(dtype=ti_dtype)
-    cg = ti.linalg.CG(A, b, x0, max_iter=50, atol=1e-6)
+    cg = ti.linalg.SparseCG(A, b, x0, max_iter=50, atol=1e-6)
     x, exit_code = cg.solve()
     res = np.linalg.solve(A_psd, b.to_numpy())
     assert exit_code == True