Skip to content
Open
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 15 additions & 13 deletions .buildkite/pipeline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ steps:
- "1.10"
- "1.11"
- "1.12"
- "1.13"
- "nightly"
adjustments:
- with:
Expand All @@ -47,7 +48,7 @@ steps:
- label: "CUDA {{matrix.cuda}}"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
- JuliaCI/julia-test#v1:
test_args: "--quickfail core base libraries"
- JuliaCI/julia-coverage#v1:
Expand All @@ -68,6 +69,7 @@ steps:
matrix:
setup:
cuda:
- "13.1"
- "13.0"
- "12.9"
- "12.8"
Expand All @@ -78,10 +80,10 @@ steps:
- "12.2"
- "12.1"
- "12.0"
adjustments:
- with:
cuda: "13.0"
soft_fail: true
#TODO adjustments:
#TODO - with:
#TODO cuda: "13.0"
#TODO soft_fail: true
commands: |
echo -e "[CUDA_Runtime_jll]\nversion = \"{{matrix.cuda}}\"" >LocalPreferences.toml

Expand Down Expand Up @@ -149,7 +151,7 @@ steps:
#- label: "NNlib.jl"
# plugins:
# - JuliaCI/julia#v1:
# version: "1.11"
# version: "1.12"
# - JuliaCI/julia-coverage#v1:
# dirs:
# - src
Expand Down Expand Up @@ -258,7 +260,7 @@ steps:
- label: "GPU-less environment"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
- JuliaCI/julia-coverage#v1:
dirs:
- src
Expand Down Expand Up @@ -290,7 +292,7 @@ steps:
- label: "Compute sanitizer"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
- JuliaCI/julia-test#v1:
test_args: "--sanitize core base"
- JuliaCI/julia-coverage#v1:
Expand All @@ -312,7 +314,7 @@ steps:
- label: "Legacy memory allocator"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
- JuliaCI/julia-test#v1:
test_args: "--quickfail core base"
- JuliaCI/julia-coverage#v1:
Expand All @@ -336,7 +338,7 @@ steps:
- label: "CuArray with {{matrix.memory}} memory"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
- JuliaCI/julia-test#v1:
test_args: "--quickfail core base libraries"
- JuliaCI/julia-coverage#v1:
Expand Down Expand Up @@ -365,7 +367,7 @@ steps:
- label: "MultiGPU"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
- JuliaCI/julia-test#v1:
test_args: "--gpu=0,1 --quickfail core base libraries"
- JuliaCI/julia-coverage#v1:
Expand All @@ -391,7 +393,7 @@ steps:
- label: ":books: Documentation"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
command: |
julia --project -e '
println("--- :julia: Instantiating project")
Expand All @@ -415,7 +417,7 @@ steps:
- label: ":racehorse: Benchmarks"
plugins:
- JuliaCI/julia#v1:
version: "1.11"
version: "1.12"
command: |
julia --project=perf -e '
using Pkg
Expand Down
26 changes: 11 additions & 15 deletions lib/nvml/NVML.jl
Original file line number Diff line number Diff line change
Expand Up @@ -12,21 +12,17 @@ import Libdl

export has_nvml

function libnvml()
@memoize begin
if Sys.iswindows()
# the NVSMI dir isn't added to PATH by the installer
nvsmi = joinpath(ENV["ProgramFiles"], "NVIDIA Corporation", "NVSMI")
if isdir(nvsmi)
joinpath(nvsmi, "nvml.dll")
else
# let's just hope for the best
"nvml"
end
else
"libnvidia-ml.so.1"
# Compile-time constant for `ccall` (Julia 1.13 needs this)
const libnvml::String = @static Sys.iswindows() ? "nvml" : "libnvidia-ml.so.1"

function __init__()
if Sys.iswindows()
# NVSMI dir isn't added to PATH by the installer; add it to Julia's DLL search path.
nvsmi = joinpath(get(ENV, "ProgramFiles", raw"C:\Program Files"), "NVIDIA Corporation", "NVSMI")
if isdir(nvsmi) && !(nvsmi in Libdl.DL_LOAD_PATH)
pushfirst!(Libdl.DL_LOAD_PATH, nvsmi)
end
end::String
end
end

function has_nvml()
Expand All @@ -37,7 +33,7 @@ function has_nvml()
return false
end

if Libdl.dlopen(libnvml(); throw_error=false) === nothing
if Libdl.dlopen(libnvml; throw_error=false) === nothing
return false
end

Expand Down
Loading
Loading