From ddde4efe24511a8809a1a06bc7944bbd84e6fef9 Mon Sep 17 00:00:00 2001 From: janEbert <janpublicebert@posteo.net> Date: Wed, 23 Oct 2024 10:58:42 +0200 Subject: [PATCH] Add PyTorch kernel cache path --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 98f5157..86806a1 100644 --- a/README.md +++ b/README.md @@ -244,6 +244,7 @@ directories mentioned here. Here are some of the variables concerning various libraries and their default values. +- PyTorch kernels: `PYTORCH_KERNEL_CACHE_PATH="$HOME"/.cache/torch/kernels` - PyTorch Hub: `TORCH_HOME="$HOME"/.cache/torch/hub` - PyTorch extensions: `TORCH_EXTENSIONS_DIR="$HOME"/.cache/torch_extensions` - Triton (PyTorch dependency): `TRITON_CACHE_DIR="$HOME"/.triton/cache` -- GitLab