From 42e9ec544a714f57b89d490e1c7a303b07f2a2b7 Mon Sep 17 00:00:00 2001
From: Miroslav Psota <9453736+MiroPsota@users.noreply.github.com>
Date: Thu, 12 Sep 2024 08:53:15 +0200
Subject: [PATCH] Add PyTorch 2.4.1

---
 README.md   | 2 +-
 get_jobs.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 6c0d663..ce0e81f 100644
--- a/README.md
+++ b/README.md
@@ -60,7 +60,7 @@ pytorch3d==0.7.6+pt2.2.1cu121
 
 ## Supported combinations
 
-Tested with PyTorch `1.11.0` - `2.3.0` and their respective compute platforms and supported OSes, with an exception for `cu102` on Windows (no VS 2017 on the GH `windows-2019` runner), and the `rocm` platform.
+Tested with PyTorch `1.11.0` - `2.4.1` and their respective compute platforms and supported OSes, with an exception for `cu102` on Windows (no VS 2017 on the GH `windows-2019` runner), and the `rocm` platform.
 
 ## Pitfalls
 
diff --git a/get_jobs.py b/get_jobs.py
index 0a59a04..c886f24 100644
--- a/get_jobs.py
+++ b/get_jobs.py
@@ -32,6 +32,7 @@ class TorchRelease:
     "2.3.0": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "rocm6.0")),
     "2.3.1": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "rocm6.0")),
     "2.4.0": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "cu124", "rocm6.1")),
+    "2.4.1": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "cu124", "rocm6.1")),
 }