working memory leak fix in tunableop

2024-06-28 15:15:12 +00:00 · 2024-06-28 15:15:12 +00:00 · 68583d3240
parent 3d50ff71b7
commit 68583d3240
1 changed files with 1 additions and 1 deletions
--- a/2
+++ b/2
@ -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"