1 files changed, 13 insertions, 3 deletions

diff --git a/epistemia/packages/machine-learning.scm b/epistemia/packages/machine-learning.scm
index 9bfb515..4f3eafb 100644
--- a/epistemia/packages/machine-learning.scm
+++ b/epistemia/packages/machine-learning.scm
@@ -16,7 +16,7 @@
   #:use-module (gnu packages rocm-libs))
 
 (define-public llama-cpp-rocm
-  (let ((tag "87589042cac2c390cec8d68fb2fad64e0a2a252a"))
+  (let ((tag "b9296"))
     (package
       (inherit llama-cpp)
       (name "llama-cpp-rocm")
@@ -29,14 +29,14 @@
                 (commit tag)))
          (file-name (git-file-name name tag))
          (sha256
-          (base32 "08q6k02rrrzizhlhfyw3bdqq03vrl6yaz8h91i5zach26ck01c3k"))))
+          (base32 "1j9icqkqaap00lhlns4j98yxd3x5h48n1izbal2i6s3vwbg0c2ib"))))
       (arguments
        (substitute-keyword-arguments (package-arguments llama-cpp)
          ((#:configure-flags flags #~'())
           #~(append #$flags
                     '(;; CMAKE_HIP_FLAGS is to workaround regression in ROCm 7.2
                       "-DCMAKE_HIP_FLAGS=\"-mllvm --amdgpu-unroll-threshold-local=600\""
-                      "-DGPU_TARGETS=gfx1100"
+                      #$(string-append "-DGPU_TARGETS=" (current-amd-gpu-targets-string))
                       ;; temporarily disable webui building until I figure out how to add second source for hf bucket
                       "-DLLAMA_USE_PREBUILT_UI=OFF" "-DLLAMA_BUILD_UI=OFF")))
          ((#:phases phases)
@@ -45,9 +45,19 @@
               (add-after 'fix-tests 'fix-tests-new
                 (lambda _
                   (substitute* "tests/CMakeLists.txt"
+                    (("llama_build_and_test\\(test-state-restore-fragmented\\.cpp .*")
+                     "")
+                    (("set_tests_properties\\(test-state-restore-fragmented PROPERTIES.*")
+                     ""))
+                  (substitute* "tests/CMakeLists.txt"
                     (("llama_build_and_test\\(test-recurrent-state-rollback\\.cpp .*")
                      "")
                     (("set_tests_properties\\(test-recurrent-state-rollback PROPERTIES.*")
+                     ""))
+                  (substitute* "tests/CMakeLists.txt"
+                    (("llama_build_and_test\\(test-save-load-state\\.cpp .*")
+                     "")
+                    (("set_tests_properties\\(test-save-load-state PROPERTIES.*")
                      ""))))
               (delete 'fix-python-shebang)))))
       (inputs