Try it out.

Author: maleadt

test/helpers/ptx.jl

@@ -61,6 +61,8 @@ function code_llvm(io::IO, @nospecialize(func), @nospecialize(types); kwargs...)
     job, kwargs = create_job(func, types; kwargs...)
     GPUCompiler.code_llvm(io, job; kwargs...)
 end
+code_llvm(@nospecialize(func), @nospecialize(types); kwargs...) =
+    code_llvm(stdout, func, types; kwargs...)
 
 function code_native(io::IO, @nospecialize(func), @nospecialize(types); kwargs...)
     job, kwargs = create_job(func, types; kwargs...)

test/ptx.jl

@@ -20,9 +20,12 @@ end
         end
     end
 
-    ir = sprint(io->PTX.code_llvm(io, mod.kernel, Tuple{mod.Aggregate}))
-    @test occursin(r"@(julia|j)_kernel\w*\(({ i64 }|\[1 x i64\])\* ", ir) ||
-          occursin(r"@(julia|j)_kernel\w*\(ptr ", ir)
+    filecheck(raw"""
+        ; TYPED: @{{(julia|j)_kernel_[0-9]+}}({{(\{ i64 \}|\[1 x i64\])}}*
+        ; OPAQUE: @{{(julia|j)_kernel_[0-9]+}}(ptr
+        """) do _
+        PTX.code_llvm(mod.kernel, Tuple{mod.Aggregate})
+    end
 
     ir = sprint(io->PTX.code_llvm(io, mod.kernel, Tuple{mod.Aggregate}; kernel=true))
     @test occursin(r"@_Z6kernel9Aggregate\(.*({ i64 }|\[1 x i64\]) ", ir)