Take path to llvm-spirv from environment

1 files changed, 6 insertions, 1 deletions

diff --git a/zluda/src/impl/module.rs b/zluda/src/impl/module.rs
index f6522f8..3635d29 100644
--- a/zluda/src/impl/module.rs
+++ b/zluda/src/impl/module.rs
@@ -148,11 +148,16 @@ impl SpirvModule {
         spirv_il: &[u8],
         ptx_lib: Option<(&'static [u8], &'static [u8])>,
     ) -> io::Result<Vec<u8>> {
+        use std::env;
         let dir = tempfile::tempdir()?;
         let mut spirv = NamedTempFile::new_in(&dir)?;
         let llvm = NamedTempFile::new_in(&dir)?;
         spirv.write_all(spirv_il)?;
-        let to_llvm_cmd = Command::new(Self::LLVM_SPIRV)
+        let llvm_spirv_path = match env::var("LLVM_SPIRV") {
+            Ok(path) => Cow::Owned(path),
+            Err(_) => Cow::Borrowed(Self::LLVM_SPIRV),
+        };
+        let to_llvm_cmd = Command::new(&*llvm_spirv_path)
             .arg("-r")
             .arg("-o")
             .arg(llvm.path())