Fix BLAS / LAPACK calls on x86

.github/workflows/CI.yml

@@ -14,7 +14,7 @@ concurrency:
   cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }}
 jobs:
   test:
-    name: ${{ matrix.test_group }}-${{ matrix.version }}
+    name: ${{ matrix.test_group }}-${{ matrix.version }}-${{ matrix.arch }}
     runs-on: ubuntu-latest
     if: github.event_name != 'schedule'
     strategy:
@@ -33,18 +33,25 @@ jobs:
           - 'integration_testing/temporalgps'
         version:
           - '1'
+        arch:
+          - x64
         include:
           - test_group: 'basic'
             version: '1.10'
+            arch: x64
           - test_group: 'integration_testing/turing'
             version: '1.10'
+            arch: x64
+          - test_group: 'basic'
+            version: '1.10'
+            arch: x86
 
     steps:
       - uses: actions/checkout@v4
       - uses: julia-actions/setup-julia@v2
         with:
           version: ${{ matrix.version }}
-          arch: x64
+          arch: ${{ matrix.arch }}
           include-all-prereleases: false
       - uses: julia-actions/cache@v2
       - uses: julia-actions/julia-buildpkg@v1

Project.toml

@@ -1,7 +1,7 @@
 name = "Mooncake"
 uuid = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 authors = ["Will Tebbutt, Hong Ge, and contributors"]
-version = "0.4.15"
+version = "0.4.16"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

src/rrules/blas.jl

@@ -1,4 +1,6 @@
-blas_name(name::Symbol) = (Symbol(name, "64_"), Symbol(BLAS.libblastrampoline))
+function blas_name(name::Symbol)
+    return (BLAS.USE_BLAS64 ? Symbol(name, "64_") : name, Symbol(BLAS.libblastrampoline))
+end
 
 function wrap_ptr_as_view(ptr::Ptr{T}, N::Int, inc::Int) where {T}
     return view(unsafe_wrap(Vector{T}, ptr, N * inc), 1:inc:N*inc)