From d4f922dd95123667e53a3b9cde566e72b21e131e Mon Sep 17 00:00:00 2001
From: Barak Michener <me@barakmich.com>
Date: Wed, 15 May 2024 20:00:17 -0700
Subject: [PATCH] fix building tch-rs with rust-analyzer

---
 .vim/nvim.lua | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.vim/nvim.lua b/.vim/nvim.lua
index e07e4e5..272c22d 100644
--- a/.vim/nvim.lua
+++ b/.vim/nvim.lua
@@ -48,6 +48,11 @@ nvim_lsp.rust_analyzer.setup({
           'cargo', 'clippy', '--workspace', '--message-format=json',
           '--all-targets', '--all-features'
         }
+      },
+      runnables = {
+        extraEnv = {
+          TORCH_CUDA_VERSION = "cu121",
+        }
       }
     }
   }