From 5cf92962cd77fcd91d193519838c20b32c4af841 Mon Sep 17 00:00:00 2001
From: Christian Guinard <28689358+christiangnrd@users.noreply.github.com>
Date: Thu, 13 Nov 2025 16:12:05 -0400
Subject: [PATCH] Update to ParallelTestRunner v2

---
 test/Project.toml |  2 +-
 test/runtests.jl  | 17 ++++++-----------
 2 files changed, 7 insertions(+), 12 deletions(-)

diff --git a/test/Project.toml b/test/Project.toml
index 7221504c2..936c644f1 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -12,4 +12,4 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]
-ParallelTestRunner = "1"
+ParallelTestRunner = "2"
diff --git a/test/runtests.jl b/test/runtests.jl
index 71522fcfd..44b3d3946 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -1,4 +1,4 @@
-using ParallelTestRunner: runtests
+using ParallelTestRunner: runtests, parse_args
 import GPUArrays
 
 include("testsuite.jl")
@@ -18,16 +18,11 @@ const init_code = quote
     end
 end
 
-custom_tests = Dict{String, Expr}()
-for AT in (:JLArray, :Array), name in keys(TestSuite.tests)
-    custom_tests["$(AT)/$name"] = :(TestSuite.tests[$name]($AT))
-end
+args = parse_args(ARGS)
 
-function test_filter(test)
-    if startswith(test, "testsuite")
-        return false
-    end
-    return true
+testsuite = Dict{String, Expr}()
+for AT in (:JLArray, :Array), name in keys(TestSuite.tests)
+    testsuite["$(AT)/$name"] = :(TestSuite.tests[$name]($AT))
 end
 
-runtests(GPUArrays, ARGS; init_code, custom_tests, test_filter)
+runtests(GPUArrays, ARGS; init_code, testsuite)