Re: PyTorch with ROCm

2024-08-11 Thread David Elsing
Hello, I updated PyTorch to 2.4.0 [1, 2] and the ROCm packages to 6.1.2. This time, I opted to put the ROCm packages into a separate channel [3] only for one version, because the packages in Guix-HPC are intended for many ROCm versions and only include versions up to 5.7.1 at the moment. The PyTo

Re: PyTorch with ROCm

2024-04-03 Thread David Elsing
Hello, Ludovic Courtès writes: > Yeah, we could think about a transformation option. Maybe > ‘--with-configure-flags=python-pytorch=-DAMDGPU_TARGETS=xyz’ would work, > and if not, we can come up with a specific transformation and/or an > procedure that takes a list of architectures and returns

Re: PyTorch with ROCm

2024-04-02 Thread Ludovic Courtès
Hello! (Cc’ing my colleague Romain who may work on related things soon.) David Elsing skribis: > It is the same as for other HIP/ROCm libraries, so the GPU architectures > chosen at build time are all available at runtime and automatically > picked. For reference, the Arch Linux package for PyT

Re: PyTorch with ROCm

2024-03-31 Thread David Elsing
Hi! Ludovic Courtès writes: > I’m happy to merge your changes in the ‘guix-hpc’ channel for the time > being (I can create you an account there if you wish so you can create > merge requests etc.). Let me know! Ok sure, that sounds good! I made the packages only for ROCm 6.0.2 so far though.

Re: PyTorch with ROCm

2024-03-28 Thread Ludovic Courtès
Hello! David Elsing skribis: > after seeing that ROCm packages [1] are available in the Guix-HPC > channel, I decided to try and package PyTorch 2.2.1 with ROCm 6.0.2. Nice! > The changes for the ROCm packages are here [4] as a modification of > Guix-HPC. There, the python-pytorch-rocm package

Re: PyTorch with ROCm

2024-03-24 Thread David Elsing
Hi Ricardo, thanks for the information! Ricardo Wurmus writes: > Oh, commit 8429f25ecd83594e80676a67ad9c54f0d6cf3f16 added > python-pytorch2 at version 2.2.1. Do you think you could adjust your > patches to modify that one instead? I already adjusted the patches yesterday to remove the python

Re: PyTorch with ROCm

2024-03-24 Thread Ricardo Wurmus
Hi David, > after seeing that ROCm packages [1] are available in the Guix-HPC > channel, I decided to try and package PyTorch 2.2.1 with ROCm 6.0.2. Excellent initiative! > For this, I first unbundled the (many) remaining dependencies of the > python-pytorch package and updated it to 2.2.1, th

PyTorch with ROCm

2024-03-24 Thread David Elsing
Hello, after seeing that ROCm packages [1] are available in the Guix-HPC channel, I decided to try and package PyTorch 2.2.1 with ROCm 6.0.2. For this, I first unbundled the (many) remaining dependencies of the python-pytorch package and updated it to 2.2.1, the patch series for which can be foun