From 6430b3e9b6b16737e8cb948fd54ca169f9dfaa12 Mon Sep 17 00:00:00 2001
From: Nazar Mokrynskyi <nazar@mokrynskyi.com>
Date: Mon, 16 Sep 2024 00:08:57 +0300
Subject: [PATCH] ROCm WIP

---
 Cargo.lock                                    | 3 +--
 shared/subspace-proof-of-space-gpu/Cargo.toml | 4 ++--
 shared/subspace-proof-of-space-gpu/build.rs   | 4 +---
 3 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 78f0dd0704b..dfc6ce76c75 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -12407,8 +12407,7 @@ dependencies = [
 [[package]]
 name = "sppark"
 version = "0.1.8"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "c55f3833d30846a26110dccb1d5366314c2c52516a9173b74238c16b24b1a9f9"
+source = "git+https://github.com/dot-asm/sppark?rev=72162c39abd90f2addf21e47e9843dbe8f10ad9d#72162c39abd90f2addf21e47e9843dbe8f10ad9d"
 dependencies = [
  "cc",
  "which",
diff --git a/shared/subspace-proof-of-space-gpu/Cargo.toml b/shared/subspace-proof-of-space-gpu/Cargo.toml
index 7456c2a1cb3..c1fb81e338f 100644
--- 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
index 0e9d891da89..56cea0c6546 100644
--- 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.