Skip to content

Commit

Permalink
test: allow the latest AMDGPU to be installed (#1116)
Browse files Browse the repository at this point in the history
  • Loading branch information
avik-pal authored Dec 3, 2024
1 parent a76d272 commit 514e554
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion lib/LuxLib/test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ if (BACKEND_GROUP == "all" || BACKEND_GROUP == "cuda")
end
end
(BACKEND_GROUP == "all" || BACKEND_GROUP == "amdgpu") &&
push!(EXTRA_PKGS, PackageSpec(; name="AMDGPU", version=v"1.0.4"))
push!(EXTRA_PKGS, PackageSpec(; name="AMDGPU"))
(BACKEND_GROUP == "all" || BACKEND_GROUP == "oneapi") &&
push!(EXTRA_PKGS, PackageSpec(; name="oneAPI"))
(BACKEND_GROUP == "all" || BACKEND_GROUP == "metal") &&
Expand Down
2 changes: 1 addition & 1 deletion lib/MLDataDevices/test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ if (BACKEND_GROUP == "all" || BACKEND_GROUP == "cuda")
end
end
(BACKEND_GROUP == "all" || BACKEND_GROUP == "amdgpu") &&
push!(EXTRA_PKGS, PackageSpec(; name="AMDGPU", version=v"1.0.4"))
push!(EXTRA_PKGS, PackageSpec(; name="AMDGPU"))
(BACKEND_GROUP == "all" || BACKEND_GROUP == "oneapi") &&
push!(EXTRA_PKGS, PackageSpec(; name="oneAPI"))
(BACKEND_GROUP == "all" || BACKEND_GROUP == "metal") &&
Expand Down
2 changes: 1 addition & 1 deletion lib/WeightInitializers/test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ const EXTRA_PKGS = PackageSpec[]
(BACKEND_GROUP == "all" || BACKEND_GROUP == "cuda") &&
push!(EXTRA_PKGS, PackageSpec("CUDA"))
(BACKEND_GROUP == "all" || BACKEND_GROUP == "amdgpu") &&
push!(EXTRA_PKGS, PackageSpec(; name="AMDGPU", version=v"1.0.4"))
push!(EXTRA_PKGS, PackageSpec(; name="AMDGPU"))
(BACKEND_GROUP == "all" || BACKEND_GROUP == "metal") &&
push!(EXTRA_PKGS, PackageSpec("Metal"))
(BACKEND_GROUP == "all" || BACKEND_GROUP == "oneapi") &&
Expand Down
2 changes: 1 addition & 1 deletion test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ if (BACKEND_GROUP == "all" || BACKEND_GROUP == "cuda")
end
end
(BACKEND_GROUP == "all" || BACKEND_GROUP == "amdgpu") &&
push!(EXTRA_PKGS, Pkg.PackageSpec(; name="AMDGPU", version=v"1.0.4"))
push!(EXTRA_PKGS, Pkg.PackageSpec(; name="AMDGPU"))

if !isempty(EXTRA_PKGS) || !isempty(EXTRA_DEV_PKGS)
@info "Installing Extra Packages for testing" EXTRA_PKGS EXTRA_DEV_PKGS
Expand Down

0 comments on commit 514e554

Please sign in to comment.