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
ure they can be combined however, as the GPU code is included >>> in the shared libraries. Thus all dependent packages like >>> python-pytorch-rocm would need to be built for each architecture as >>> well, which is a large duplication for the non-GPU parts. >> >&g

Re: PyTorch with ROCm

2024-03-31 Thread David Elsing
ot sure they can be combined however, as the GPU code is included >> in the shared libraries. Thus all dependent packages like >> python-pytorch-rocm would need to be built for each architecture as >> well, which is a large duplication for the non-GPU parts. > > Yeah, but maybe that

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-

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

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

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