gemm logs (#403)

* gemm logs * bump version
worldcoin · Sep 18, 2024 · e550fc2 · e550fc2
1 parent ab64d5a
commit e550fc2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/deploy/stage/common-values-iris-mpc.yaml b/deploy/stage/common-values-iris-mpc.yaml
@@ -1,4 +1,4 @@
-image: "ghcr.io/worldcoin/iris-mpc:v0.6.8"
+image: "ghcr.io/worldcoin/iris-mpc:v0.6.9"
 
 environment: stage
 replicaCount: 1

diff --git a/iris-mpc-gpu/src/dot/share_db.rs b/iris-mpc-gpu/src/dot/share_db.rs
@@ -70,6 +70,21 @@ pub fn gemm(
     alpha: i32,
     beta: i32,
 ) {
+    tracing::info!(
+        "gemm: a: {:?}, b: {:?}, c: {:?}, a_offset: {}, b_offset: {}, c_offset: {}, m: {}, n: {}, \
+         k: {}, alpha: {}, beta: {}",
+        a,
+        b,
+        c,
+        a_offset,
+        b_offset,
+        c_offset,
+        m,
+        n,
+        k,
+        alpha,
+        beta
+    );
     // https://docs.nvidia.com/cuda/cublas/#cublasgemmex:
     // "CUBLAS_COMPUTE_32I and CUBLAS_COMPUTE_32I_PEDANTIC compute types are only supported with A, B being 4-byte aligned and lda, ldb being multiples of 4."
     assert!(m % 4 == 0, "m must be a multiple of 4");