From 68583d3240f0dfed38ffe290ed3205a778cbe189 Mon Sep 17 00:00:00 2001 From: Felix Marty <9808326+fxmarty@users.noreply.github.com> Date: Fri, 28 Jun 2024 15:15:12 +0000 Subject: [PATCH] working memory leak fix in tunableop --- Dockerfile_amd | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile_amd b/Dockerfile_amd index 56a6ac5d..7eb80ebf 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -98,7 +98,7 @@ RUN pip uninstall -y triton && \ RUN git clone --depth 1 --recursive --single-branch --branch 2.3-patched https://github.com/fxmarty/pytorch.git pytorch && \ cd pytorch && \ - git checkout adbb514d024d4ba1b6f9c126d321fd5e7597e966 && \ + git checkout ceaa1e4a7b66e818ea4e56925bb4a5dff8c56055 && \ pip install -r requirements.txt --no-cache-dir ARG _GLIBCXX_USE_CXX11_ABI="1"