From 0b074ddaf35b4acebf43e96ebd7d1c293873e28c Mon Sep 17 00:00:00 2001 From: GideokKim Date: Thu, 8 Aug 2024 15:19:07 +0900 Subject: [PATCH] build: add `gpu_heavy_test` tag Currently, NTT operates assuming that GPU memory is always sufficient. Until the issue is resolved, add a tag to manually test in cuda tests. --- .bazelrc | 4 ++++ tachyon/zk/plonk/examples/BUILD.bazel | 5 +++++ tachyon/zk/plonk/examples/fibonacci/BUILD.bazel | 3 +++ 3 files changed, 12 insertions(+) diff --git a/.bazelrc b/.bazelrc index 32fb813cc..8ba9bfc11 100644 --- a/.bazelrc +++ b/.bazelrc @@ -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 diff --git a/tachyon/zk/plonk/examples/BUILD.bazel b/tachyon/zk/plonk/examples/BUILD.bazel index c05413380..9ab392418 100644 --- a/tachyon/zk/plonk/examples/BUILD.bazel +++ b/tachyon/zk/plonk/examples/BUILD.bazel @@ -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", @@ -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", @@ -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", @@ -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", @@ -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", diff --git a/tachyon/zk/plonk/examples/fibonacci/BUILD.bazel b/tachyon/zk/plonk/examples/fibonacci/BUILD.bazel index 0d38faf5f..6e9c33236 100644 --- a/tachyon/zk/plonk/examples/fibonacci/BUILD.bazel +++ b/tachyon/zk/plonk/examples/fibonacci/BUILD.bazel @@ -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", @@ -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", @@ -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",