ci: Fixes for pre-built wheels#214
Conversation
Signed-off-by: oliver könig <okoenig@nvidia.com>
Signed-off-by: oliver könig <okoenig@nvidia.com>
Signed-off-by: oliver könig <okoenig@nvidia.com>
Signed-off-by: oliver könig <okoenig@nvidia.com>
Signed-off-by: oliver könig <okoenig@nvidia.com>
|
Here's a passing run: https://github.com/ko3n1g/DeepGEMM/actions/runs/18443482987 |
|
After merging this one, can we create and push a new tag @LyricZhao ? |
Sure! Can you bring Python 3.8 back? |
which cuda and torch version do you expect/ want to support with py3.8? |
|
How about CUDA 12.3 + PyTorch 2.1.0? |
Signed-off-by: oliver könig <okoenig@nvidia.com>
Signed-off-by: oliver könig <okoenig@nvidia.com>
|
Okay, looks like we've got it now. Here's the job that also builds a wheel for Py3.8/Cu12.3,PyT2.1.0: https://github.com/ko3n1g/DeepGEMM/actions/runs/18457600231/job/52581827327 Since support for Py3.8 is quite limited I didn't include a vast grid but just this singular configuration |
|
I think we need to use cuda11.8 instead: https://github.com/pytorch/pytorch/blob/main/RELEASE.md#release-compatibility-matrix |
Signed-off-by: oliver könig <okoenig@nvidia.com>
|
@LyricZhao Finally got it working with cuda12.1: https://github.com/ko3n1g/DeepGEMM/actions/runs/18465627532/job/52608861167 |
|
LGTM, ready for merge now? |
|
Yep! |
2 participants
This brings support for NGC PyTorch containers and removes cuda11 since it doesn't seem to be compatible anymore