Skip to content

Commit 1dafd89

Browse files
no gpu testing
1 parent 0390aee commit 1dafd89

File tree

2 files changed

+5
-11
lines changed

2 files changed

+5
-11
lines changed
Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
using TestItemRunner
2-
using Pkg
3-
# See https://www.julia-vscode.org/docs/stable/userguide/testitems/
4-
# for how to run the tests within VS Code.
5-
# See test_module.jl for the test infrasructure.
2+
3+
## See https://www.julia-vscode.org/docs/stable/userguide/testitems/
4+
## for how to run the tests within VS Code.
5+
## See test_module.jl for the test infrastructure.
66

77
## Uncomment below to change the default test settings
88
# ENV["GNN_TEST_CPU"] = "false"
@@ -14,18 +14,12 @@ if get(ENV, "GNN_TEST_CPU", "true") == "true"
1414
@run_package_tests filter = ti -> :gpu ti.tags
1515
end
1616
if get(ENV, "GNN_TEST_CUDA", "false") == "true"
17-
Pkg.add(["CUDA", "cuDNN"])
18-
using CUDA
1917
@run_package_tests filter = ti -> :gpu ti.tags
2018
end
2119
if get(ENV, "GNN_TEST_AMDGPU", "false") == "true"
22-
Pkg.add("AMDGPU")
23-
using AMDGPU
2420
@run_package_tests filter = ti -> :gpu ti.tags
2521
end
2622
if get(ENV, "GNN_TEST_Metal", "false") == "true"
27-
Pkg.add("Metal")
28-
using Metal
2923
@run_package_tests filter = ti -> :gpu ti.tags
3024
end
3125

GraphNeuralNetworks/test/test_module.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ using Pkg
1111

1212
## Uncomment below to change the default test settings
1313
# ENV["GNN_TEST_CPU"] = "false"
14-
ENV["GNN_TEST_CUDA"] = "true"
14+
# ENV["GNN_TEST_CUDA"] = "true"
1515
# ENV["GNN_TEST_AMDGPU"] = "true"
1616
# ENV["GNN_TEST_Metal"] = "true"
1717

0 commit comments

Comments
 (0)