From fbee3e0076c02af6da01304000917ef46f85c5c9 Mon Sep 17 00:00:00 2001
From: Christian Guinard <28689358+christiangnrd@users.noreply.github.com>
Date: Wed, 4 Feb 2026 12:58:45 -0400
Subject: [PATCH 1/2] ParallelTestRunner 2.2

---
 test/Project.toml |  2 +-
 test/runtests.jl  | 16 ++++++++++++----
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/test/Project.toml b/test/Project.toml
index 00d4019cd..67b5f0b86 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -21,4 +21,4 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]
-ParallelTestRunner = "2"
+ParallelTestRunner = "2.2"
diff --git a/test/runtests.jl b/test/runtests.jl
index 9426a70f9..875503971 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -96,16 +96,16 @@ if filter_tests!(testsuite, args)
 end
 
 # workers to run tests on
-function test_worker(name)
+function test_worker(name, init_worker_code)
     if name == "capturing"
-        return addworker(env=["METAL_CAPTURE_ENABLED"=>"1"])
+        return addworker(; env=["METAL_CAPTURE_ENABLED"=>"1"], init_worker_code)
     end
 
     return nothing
 end
 
 # code to run in each test's sandbox module before running the test
-init_code = quote
+init_worker_code = quote
     using Metal, Adapt, ObjectiveC, ObjectiveC.Foundation, BFloat16s
 
     # XXX: expose this as --validate
@@ -158,4 +158,12 @@ init_code = quote
     end
 end
 
-runtests(Metal, args; testsuite, init_code, test_worker)
+init_code = quote
+    using Metal, Adapt, ObjectiveC, ObjectiveC.Foundation, BFloat16s
+
+    # bring used symbols into the temporary module
+    import ..TestSuite, ..testf
+    import ..runtime_validation, ..shader_validation, ..capturing, ..@grab_output, ..@on_device
+end
+
+runtests(Metal, args; testsuite, init_code, init_worker_code, test_worker)

From 0b378c4b5f1c98db8930198d5e2795bb95e04f90 Mon Sep 17 00:00:00 2001
From: Christian Guinard <28689358+christiangnrd@users.noreply.github.com>
Date: Wed, 4 Feb 2026 12:59:03 -0400
Subject: [PATCH 2/2] [Temp]

---
 test/Project.toml | 3 ++-
 test/runtests.jl  | 3 +++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/test/Project.toml b/test/Project.toml
index 67b5f0b86..675ffaafa 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -11,6 +11,7 @@ LLVM = "929cbde3-209d-540e-8aea-75f648917ca0"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 ObjectiveC = "e86c9b32-1129-44ac-8ea0-90d5bb39ded9"
 ParallelTestRunner = "d3525ed8-44d0-4b2c-a655-542cee43accc"
+Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 ScopedValues = "7e506255-f358-4e82-b7e4-beb19740aa63"
@@ -21,4 +22,4 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]
-ParallelTestRunner = "2.2"
+ParallelTestRunner = "2"
diff --git a/test/runtests.jl b/test/runtests.jl
index 875503971..989bd06e2 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -1,3 +1,6 @@
+using Pkg
+Pkg.add(url="https://github.com/christiangnrd/ParallelTestRunner.jl", rev="total_time")
+
 using Metal
 using ParallelTestRunner