Hotfix: Fix offset calculation to prevent overflow if offset is really large

clients/gtest/trsm_gtest.yaml

@@ -60,6 +60,15 @@ Definitions:
   - &large_memory_matrix_size_range
     - { M: 8320, N: 128, lda: 8320, ldb: 8320 }
 
+  - &size_t_left_matrix_size_range
+#   - { M: 4, N: 46435, lda: 4, ldb: 46435 }
+    - { M: 46345, N: 4, lda: 46345, ldb: 46345 }
+#   - { M: 47000, N: 4, lda: 47000, ldb: 47000 } # calls rocblas_internal_gemm_template with batch_count=367, stride_a=6016128
+
+  - &size_t_right_matrix_size_range
+    - { M: 4, N: 46345, lda: 46345, ldb: 4 }
+#   - { M: 4, N: 47000, lda: 47000, ldb: 4 }  # calls rocblas_internal_gemm_template with batch_count=367, stride_a=6016128
+
   - &substitution_size_range_thorough
     - { M:   1, N:  1, lda: 100, ldb: 100 }
     - { M:   1, N: 32, lda: 100, ldb: 100 }
@@ -486,6 +495,24 @@ Tests:
   matrix_size: *testset2_matrix_size_range
   alpha: [ 1 ]
 
+- name: trsm_size_t_left
+  category: nightly
+  function: trsm
+  precision: *single_precision
+  arguments:
+    - { side: L, uplo: L, transA: N, diag: N }
+  matrix_size: *size_t_left_matrix_size_range
+  alpha: [2]
+
+- name: trsm_size_t_right
+  category: nightly
+  function: trsm
+  precision: *single_precision
+  arguments:
+    - { side: R, uplo: L, transA: N, diag: N }
+  matrix_size: *size_t_right_matrix_size_range
+  alpha: [2]
+
 - name: trsm_large
   category: nightly
   function: trsm

library/src/blas3/Tensile/gemm_tensile.hpp

@@ -139,19 +139,33 @@ inline rocblas_status call_tensile(rocblas_handle    handle,
     }
 #endif
 
-    RocblasContractionProblem<T> problem{handle,   trans_a,
-                                         trans_b,  m,
-                                         n,        k,
-                                         alpha,    A,
-                                         nullptr,  ld_a,
-                                         stride_a, offset_a,
-                                         B,        nullptr,
-                                         ld_b,     stride_b,
-                                         offset_b, beta,
-                                         C,        nullptr,
-                                         ld_c,     stride_c,
-                                         offset_c, batch_count,
-                                         true,     rocblas_gemm_flags_none};
+    // pre apply offsets for non-batched and strided
+    RocblasContractionProblem<T> problem{handle,
+                                         trans_a,
+                                         trans_b,
+                                         m,
+                                         n,
+                                         k,
+                                         alpha,
+                                         A + offset_a,
+                                         nullptr,
+                                         ld_a,
+                                         stride_a,
+                                         0 /* offset_a */,
+                                         B + offset_b,
+                                         nullptr,
+                                         ld_b,
+                                         stride_b,
+                                         0 /* offset_b */,
+                                         beta,
+                                         C + offset_c,
+                                         nullptr,
+                                         ld_c,
+                                         stride_c,
+                                         0 /* offset_c */,
+                                         batch_count,
+                                         true,
+                                         rocblas_gemm_flags_none};
 
     return runContractionProblem(problem);
 }