Skip to content

Commit

Permalink
fixup! Use PrecompileTools to warmup CUDA.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
vchuravy committed Apr 19, 2024
1 parent d27e5eb commit c7f880c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/precompile.jl
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,8 @@ precompile(cudaconvert, (Function,))
precompile(Core.kwfunc(cudacall), (NamedTuple{(:threads, :blocks), Tuple{Int64, Int64}},typeof(cudacall),CuFunction,Type{Tuple{}}))
precompile(Core.kwfunc(launch), (NamedTuple{(:threads, :blocks), Tuple{Int64, Int64}},typeof(launch),CuFunction))

@static if VERSION >= v"1.11.0-DEV.1603"
using PrecompileTools: @setup_workload, @compile_workload
@static if VERSION >= v"1.11.0-DEV.1603"
@setup_workload let
@compile_workload begin
target = PTXCompilerTarget(; cap=v"7.5")
Expand Down

0 comments on commit c7f880c

Please sign in to comment.