ZLUDA 3.8.5 (#56)

* Enable runtime compiler for PyTorch support on Windows. * Bump version.
lshqqytiger · Jan 3, 2025 · 2930436 · 2930436
1 parent 1b6e012
commit 2930436
Show file tree

Hide file tree

Showing 13 changed files with 482 additions and 491 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -14,7 +14,6 @@ members = [
     "hipblaslt-sys",
     "hipfft-sys",
     "hiprt-sys",
-    "hiprtc-sys",
     "miopen-sys",
     "offline_compiler",
     "optix_base",

diff --git a/README.md b/README.md
@@ -239,16 +239,7 @@ Performance is currently much lower than the native HIP backend, see the discuss
   torch.backends.cuda.enable_mem_efficient_sdp(False)
   ```
 
-  If you are getting an error about nvrtc/hiprtc, insert
-
-  ```py
-  def jit_script(f, *_, **__):
-      f.graph = torch._C.Graph() # pylint: disable=protected-access
-      return f
-  torch.jit.script = jit_script
-  ```
-
-  You may have an issue while running `torch.topk`. If so, insert
+  If you have an issue while running `torch.topk`, insert the codes below
 
   ```py
   _topk = torch.topk

diff --git a/hiprtc-sys/Cargo.toml b/hiprtc-sys/Cargo.toml
diff --git a/hiprtc-sys/README b/hiprtc-sys/README
diff --git a/hiprtc-sys/build.rs b/hiprtc-sys/build.rs
diff --git a/hiprtc-sys/src/hiprtc.rs b/hiprtc-sys/src/hiprtc.rs
diff --git a/hiprtc-sys/src/lib.rs b/hiprtc-sys/src/lib.rs
diff --git a/zluda_inject/src/bin.rs b/zluda_inject/src/bin.rs
@@ -83,7 +83,7 @@ pub fn main_impl() -> Result<(), Box<dyn Error>> {
         match argument.to_str() {
             Some(argument) => match argument {
                 "--version" => {
-                    println!("ZLUDA 3.8.4");
+                    println!("ZLUDA 3.8.5");
                     process::exit(0);
                 }
                 "--" => break,

diff --git a/zluda_rtc/Cargo.toml b/zluda_rtc/Cargo.toml
@@ -8,12 +8,9 @@ edition = "2018"
 name = "nvrtc"
 crate-type = ["cdylib"]
 
-[features]
-rocm5 = ["hip_common/rocm5"]
-
 [dependencies]
-hip_common = { path = "../hip_common" }
-hiprtc-sys = { path = "../hiprtc-sys" }
+libloading = "0.8"
+lazy_static = "1.4"
 
 [package.metadata.zluda]
 linux_names = ["libnvrtc.so.10", "libnvrtc.so.11"]

diff --git a/zluda_rtc/README b/zluda_rtc/README
@@ -1,3 +1,3 @@
-bindgen include/nvrtc.h -o src/nvrtc.rs --allowlist-function="^nvrtc.*" --default-enum-style=newtype --no-layout-tests --no-derive-debug -- -Iinclude
+bindgen include/nvrtc.h -o src/nvrtc.rs --allowlist-function="^nvrtc.*" --default-enum-style=newtype --no-layout-tests --no-derive-debug --dynamic-loading LibNvrtc --dynamic-link-require-all -- -Iinclude
 sed -i -e 's/extern "C" {//g' -e 's/-> nvrtcResult;/-> nvrtcResult { crate::unsupported()/g' -e 's/pub fn /#[no_mangle] pub extern "system" fn /g' src/nvrtc.rs
 rustfmt src/nvrtc.rs