Switch to ReTestItems.jl.

Author: maleadt

test/Project.toml

@@ -4,6 +4,7 @@ InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 LLVM = "929cbde3-209d-540e-8aea-75f648917ca0"
 Metal_LLVM_Tools_jll = "0418c028-ff8c-56b8-a53e-0f9676ed36fc"
 REPL = "3fa0cd96-eef1-5676-8a61-b3b8758bbffb"
+ReTestItems = "817f1d60-ba6b-4fd5-9520-3cf149f6a823"
 SPIRV_LLVM_Translator_unified_jll = "85f0d8ed-5b39-5caa-b1ae-7472de402361"
 SPIRV_Tools_jll = "6ac6d60f-d740-5983-97d7-a4482c0689f4"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"

test/bpf.jl renamed to test/bpf_tests.jl

@@ -1,45 +1,43 @@
-@testset "eBPF" begin
-
-include("definitions/bpf.jl")
+@testitem "BPF" setup=[BPF, Helpers] begin
 
 ############################################################################################
 
 @testset "No-op" begin
     kernel() = 0
 
-    output = sprint(io->bpf_code_native(io, kernel, ()))
+    output = sprint(io->BPF.code_native(io, kernel, ()))
     @test occursin("\tr0 = 0\n\texit", output)
 end
 @testset "Return argument" begin
     kernel(x) = x
 
-    output = sprint(io->bpf_code_native(io, kernel, (UInt64,)))
+    output = sprint(io->BPF.code_native(io, kernel, (UInt64,)))
     @test occursin("\tr0 = r1\n\texit", output)
 end
 @testset "Addition" begin
     kernel(x) = x+1
 
-    output = sprint(io->bpf_code_native(io, kernel, (UInt64,)))
+    output = sprint(io->BPF.code_native(io, kernel, (UInt64,)))
     @test occursin("\tr0 = r1\n\tr0 += 1\n\texit", output)
 end
 @testset "Errors" begin
     kernel(x) = fakefunc(x)
 
-    @test_throws GPUCompiler.InvalidIRError bpf_code_execution(kernel, (UInt64,))
+    @test_throws GPUCompiler.InvalidIRError BPF.code_execution(kernel, (UInt64,))
 end
 @testset "Function Pointers" begin
     @testset "valid" begin
         goodcall(x) = Base.llvmcall("%2 = call i64 inttoptr (i64 3 to i64 (i64)*)(i64 %0)\nret i64 %2", Int, Tuple{Int}, x)
         kernel(x) = goodcall(x)
 
-        output = sprint(io->bpf_code_native(io, kernel, (Int,)))
+        output = sprint(io->BPF.code_native(io, kernel, (Int,)))
         @test occursin("\tcall 3\n\texit", output)
     end
     @testset "invalid" begin
         badcall(x) = Base.llvmcall("%2 = call i64 inttoptr (i64 3000 to i64 (i64)*)(i64 %0)\nret i64 %2", Int, Tuple{Int}, x)
         kernel(x) = badcall(x)
 
-        @test_throws GPUCompiler.InvalidIRError bpf_code_execution(kernel, (Int,))
+        @test_throws GPUCompiler.InvalidIRError BPF.code_execution(kernel, (Int,))
     end
 end
 

test/bpf_testsetup.jl

@@ -0,0 +1,39 @@
+@testsetup module BPF
+
+using GPUCompiler
+
+
+# create a native test compiler, and generate reflection methods for it
+
+include("runtime.jl")
+struct CompilerParams <: AbstractCompilerParams end
+GPUCompiler.runtime_module(::CompilerJob{<:Any,CompilerParams}) = TestRuntime
+
+function create_job(@nospecialize(func), @nospecialize(types);
+                    kernel::Bool=false, always_inline=false, kwargs...)
+    source = methodinstance(typeof(func), Base.to_tuple_type(types), Base.get_world_counter())
+    target = BPFCompilerTarget()
+    params = CompilerParams()
+    config = CompilerConfig(target, params; kernel, always_inline)
+    CompilerJob(source, config), kwargs
+end
+
+function code_llvm(io::IO, @nospecialize(func), @nospecialize(types); kwargs...)
+    job, kwargs = create_job(func, types; kwargs...)
+    GPUCompiler.code_llvm(io, job; kwargs...)
+end
+
+function code_native(io::IO, @nospecialize(func), @nospecialize(types); kwargs...)
+    job, kwargs = create_job(func, types; kwargs...)
+    GPUCompiler.code_native(io, job; kwargs...)
+end
+
+# simulates codegen for a kernel function: validates by default
+function code_execution(@nospecialize(func), @nospecialize(types); kwargs...)
+    job, kwargs = create_job(func, types; kwargs...)
+    JuliaContext() do ctx
+        GPUCompiler.compile(:asm, job; kwargs...)
+    end
+end
+
+end

test/definitions/bpf.jl

@@ -1,4 +1,4 @@
-@testset "examples" begin
+@testitem "examples" begin
 
 function find_sources(path::String, sources=String[])
     if isdir(path)
@@ -11,13 +11,13 @@ function find_sources(path::String, sources=String[])
     sources
 end
 
-examples_dir = joinpath(@__DIR__, "..", "examples")
-examples = find_sources(examples_dir)
-filter!(file -> readline(file) != "# EXCLUDE FROM TESTING", examples)
-filter!(file -> !occursin("Kaleidoscope", file), examples)
+dir = joinpath(@__DIR__, "..", "examples")
+files = find_sources(dir)
+filter!(file -> readline(file) != "# EXCLUDE FROM TESTING", files)
+filter!(file -> !occursin("Kaleidoscope", file), files)
 
-cd(examples_dir) do
-    examples = relpath.(examples, Ref(examples_dir))
+cd(dir) do
+    examples = relpath.(files, Ref(dir))
     @testset for example in examples
         cmd = `$(Base.julia_cmd()) --project=$(Base.active_project())`
         @test success(pipeline(`$cmd $example`, stderr=stderr))