Test s390x in cross tests

bluss · Jul 27, 2024 · 34e7f6e · 34e7f6e
1 parent c7ab1ac
commit 34e7f6e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -10,6 +10,7 @@ env:
   CARGO_TERM_COLOR: always
   CARGO_INCREMENTAL: 0
   MATMUL_NUM_THREADS: 4
+  RUST_BACKTRACE: full
 
 jobs:
   tests:
@@ -119,6 +120,9 @@ jobs:
     strategy:
       matrix:
         include:
+          - rust: stable
+            target: s390x-unknown-linux-gnu
+            features: constconf cgemm threading
           - rust: stable
             target: aarch64-unknown-linux-gnu
             features: constconf cgemm threading
@@ -144,9 +148,8 @@ jobs:
         run: cargo install cross
       - name: Tests
         run: cross test --target "${{ matrix.target }}" --features "${{ matrix.features }}"
-        env:
-          MMTEST_FAST_TEST: 1
-          RUSTFLAGS: -Copt-level=2
+      - name: Tests
+        run: cross test --release --target "${{ matrix.target }}" --features "${{ matrix.features }}"
 
   cargo-careful:
     runs-on: ubuntu-latest

diff --git a/src/gemm.rs b/src/gemm.rs
@@ -394,7 +394,7 @@ unsafe fn gemm_packed<K>(nc: usize, kc: usize, mc: usize,
             {
                 ptr = MASK_BUF.with(|buf| (*buf.get()).buffer.as_mut_ptr());
             }
-            ptr = align_ptr(K::align_to(), ptr);
+            ptr = align_ptr(32, ptr);
             slice::from_raw_parts_mut(ptr as *mut K::Elem, KERNEL_MAX_SIZE / size_of::<K::Elem>())
         })
         .for_each(move |_tp, mask_buf, l2, nr_| {