Skip to content

Commit

Permalink
build: add gpu_heavy_test tag
Browse files Browse the repository at this point in the history
Currently, NTT operates assuming that GPU memory is always sufficient. Until
the issue is resolved, add a tag to manually test in cuda tests.
  • Loading branch information
GideokKim committed Aug 8, 2024
1 parent 0e0c473 commit 0b074dd
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 0 deletions.
4 changes: 4 additions & 0 deletions .bazelrc
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,10 @@ build:cuda --define absl=0
build --//:has_exception

test --test_tag_filters -benchmark,-manual,-cuda,-rust
# TODO(GideokKim): Currently, NTT operates assuming that GPU memory is always
# sufficient. Until the issue is resolved, add a tag to manually test in cuda
# tests.
test:cuda --test_tag_filters -benchmark,-manual,-cuda,-rust,-gpu_heavy_test

build:linux --build_tag_filters -objc,-cuda,-rust
build:macos --build_tag_filters -cuda,-rust,-doxygen
Expand Down
5 changes: 5 additions & 0 deletions tachyon/zk/plonk/examples/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -146,6 +146,7 @@ tachyon_cc_unittest(
name = "multi_lookup_circuit_test",
srcs = ["multi_lookup_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
":multi_lookup_circuit",
":multi_lookup_circuit_test_data",
Expand All @@ -156,6 +157,7 @@ tachyon_cc_unittest(
name = "shuffle_circuit_test",
srcs = ["shuffle_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
":shuffle_circuit",
":shuffle_circuit_test_data",
Expand All @@ -166,6 +168,7 @@ tachyon_cc_unittest(
name = "shuffle_api_circuit_test",
srcs = ["shuffle_api_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
":shuffle_api_circuit",
":shuffle_api_circuit_test_data",
Expand All @@ -176,6 +179,7 @@ tachyon_cc_unittest(
name = "simple_circuit_test",
srcs = ["simple_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
":simple_circuit",
":simple_circuit_test_data",
Expand All @@ -186,6 +190,7 @@ tachyon_cc_unittest(
name = "simple_lookup_circuit_test",
srcs = ["simple_lookup_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
":simple_lookup_circuit",
":simple_lookup_circuit_test_data",
Expand Down
3 changes: 3 additions & 0 deletions tachyon/zk/plonk/examples/fibonacci/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -75,6 +75,7 @@ tachyon_cc_unittest(
name = "fibonacci1_circuit_test",
srcs = ["fibonacci1_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
"fibonacci1_circuit_test_data",
":fibonacci1_circuit",
Expand All @@ -85,6 +86,7 @@ tachyon_cc_unittest(
name = "fibonacci2_circuit_test",
srcs = ["fibonacci2_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
"fibonacci2_circuit_test_data",
":fibonacci2_circuit",
Expand All @@ -95,6 +97,7 @@ tachyon_cc_unittest(
name = "fibonacci3_circuit_test",
srcs = ["fibonacci3_circuit_test.cc"],
env = tachyon_openmp_num_threads_env(4),
tags = ["gpu_heavy_test"],
deps = COMMON_TEST_DEPS + [
"fibonacci3_circuit_test_data",
":fibonacci3_circuit",
Expand Down

0 comments on commit 0b074dd

Please sign in to comment.