Skip to content

Commit

Permalink
enable offlinecach
Browse files Browse the repository at this point in the history
  • Loading branch information
@FantasyVR
FantasyVR committed Nov 11, 2022
1 parent 26698e0 commit b6f938f
Show file tree
Hide file tree
Showing 3 changed files with 17 additions and 17 deletions.
2 changes: 1 addition & 1 deletion misc/sparse_matrix.py
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import taichi as ti

ti.init(arch=ti.x64, debug=True, offline_cache=False)
ti.init(arch=ti.x64)

n = 8

Expand Down
26 changes: 13 additions & 13 deletions tests/python/test_sparse_matrix.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_builder_deprecated_anno(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand All @@ -34,7 +34,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_builder(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand All @@ -59,7 +59,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_build_sparse_matrix_frome_ndarray(dtype, storage_format):
n = 8
triplets = ti.Vector.ndarray(n=3, dtype=ti.f32, shape=n)
Expand All @@ -85,7 +85,7 @@ def fill(triplets: ti.types.ndarray()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_shape(dtype, storage_format):
n, m = 8, 9
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand All @@ -108,7 +108,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_element_access(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand All @@ -132,7 +132,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_element_modify(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand All @@ -156,7 +156,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_addition(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down Expand Up @@ -190,7 +190,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder(),
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_subtraction(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down Expand Up @@ -224,7 +224,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder(),
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_scalar_multiplication(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand All @@ -250,7 +250,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_transpose(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand All @@ -276,7 +276,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_elementwise_multiplication(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down Expand Up @@ -310,7 +310,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder(),
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_multiplication(dtype, storage_format):
n = 2
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down Expand Up @@ -345,7 +345,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder(),
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_nonsymmetric_multiplication(dtype, storage_format):
n, k, m = 2, 3, 4
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down
6 changes: 3 additions & 3 deletions tests/python/test_spmv.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_vector_multiplication1(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down Expand Up @@ -37,7 +37,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder(), b: ti.template()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_vector_multiplication2(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down Expand Up @@ -69,7 +69,7 @@ def fill(Abuilder: ti.types.sparse_matrix_builder(), b: ti.template()):
(ti.f32, 'row_major'),
(ti.f64, 'col_major'),
(ti.f64, 'row_major')])
@test_utils.test(arch=ti.cpu, offline_cache=False)
@test_utils.test(arch=ti.cpu)
def test_sparse_matrix_vector_multiplication3(dtype, storage_format):
n = 8
Abuilder = ti.linalg.SparseMatrixBuilder(n,
Expand Down

0 comments on commit b6f938f

Please sign in to comment.