Skip to content
This repository has been archived by the owner on Jul 24, 2024. It is now read-only.
/ LuxAMDGPU.jl Public archive

Trigger Package for AMDGPU Support in Lux.jl

License

Notifications You must be signed in to change notification settings

LuxDL/LuxAMDGPU.jl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

37 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LuxAMDGPU

Join the chat at https://julialang.zulipchat.com #machine-learning Latest Docs Stable Docs

CI Buildkite NVIDIA GPU CI codecov Package Downloads

ColPrac: Contributor's Guide on Collaborative Practices for Community Packages SciML Code Style

Caution

LuxAMDPU.jl was used previously as a means to load AMDGPU specific extensions in Lux.jl. However in most recent versions of Lux.jl, these functionalities can be directly accessed by simply loading AMDGPU.jl. This package is now deprecated.

To use the LuxAMDGPU.functional() functionality, use MLDataDevices.functional(LuxAMDGPUDevice).

If you are not using LuxAMDGPU.jl as a direct dependency, please consider opening an issue on any packages you are using that do use it as a dependency. From Julia 1.9 onwards, you can query ]why LuxAMDGPU to figure out which package originally brings it in as a dependency.

LuxAMDGPU is meant to be used as a trigger package for all AMDGPU dependencies in Lux. Users requiring AMDGPU support should install LuxAMDGPU and load it alongside Lux.

Installation

LuxAMDGPU is a registered package, and can be installed using the Julia package manager:

] add LuxAMDGPU