ROCm WIP

autonomys · Sep 15, 2024 · 6430b3e · 6430b3e
1 parent 1d5ac2a
commit 6430b3e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/shared/subspace-proof-of-space-gpu/Cargo.toml b/shared/subspace-proof-of-space-gpu/Cargo.toml
@@ -16,8 +16,8 @@ include = [
 blst = { version = "0.3.13", optional = true }
 rust-kzg-blst = { git = "https://github.com/grandinetech/rust-kzg", rev = "6c8fcc623df3d7e8c0f30951a49bfea764f90bf4", default-features = false, optional = true }
 # TODO: This is `rocm` branch, it is needed for ROCm support
-#sppark = { git = "https://github.com/dot-asm/sppark", rev = "8eeafe0f6cc0ca8211b1be93922df1b5a118bbd2", optional = true }
-sppark = { version = "0.1.8", optional = true }
+sppark = { git = "https://github.com/dot-asm/sppark", rev = "72162c39abd90f2addf21e47e9843dbe8f10ad9d", optional = true }
+#sppark = { version = "0.1.8", optional = true }
 subspace-core-primitives = { version = "0.1.0", path = "../../crates/subspace-core-primitives", default-features = false, optional = true }
 
 [dev-dependencies]

diff --git a/shared/subspace-proof-of-space-gpu/build.rs b/shared/subspace-proof-of-space-gpu/build.rs
@@ -23,9 +23,7 @@ fn main() {
         if cfg!(debug_assertions) {
             hipcc.opt_level(1);
         }
-        hipcc.flag("--offload-arch=native,gfx1100,gfx1030,gfx942,gfx90a,gfx908");
-        // 6 corresponds to the number of offload-arch
-        hipcc.flag("-parallel-jobs=6");
+        hipcc.flag("--offload-arch=gfx1100,gfx1030,gfx942,gfx90a,gfx908");
         // This controls how error strings get handled in the FFI. When defined error strings get
         // returned from the FFI, and Rust must then free them. When not defined error strings are
         // not returned.