gpu test conditional on TEST_GPU

emer · Dec 20, 2024 · 41be803 · 41be803
1 parent 65288d1
commit 41be803
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/axon/basic_test.goal b/axon/basic_test.goal
@@ -411,9 +411,9 @@ func TestNetActShort(t *testing.T) {
 }
 
 func TestGPUAct(t *testing.T) {
-	// if os.Getenv("TEST_GPU") != "true" {
-	// 	t.Skip("Set TEST_GPU env var to run GPU tests")
-	// }
+	if os.Getenv("TEST_GPU") != "true" {
+		t.Skip("Set TEST_GPU env var to run GPU tests")
+	}
 	NetActTestShort(t, Tol6, true)
 	// NetActTest(t, Tol6, true, true)
 }