small update per issue #158

xiaoyeli · Jan 5, 2025 · 6574d1f · 6574d1f
1 parent eac44cf
commit 6574d1f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/SRC/complex16/zSchCompUdt-gpu.c b/SRC/complex16/zSchCompUdt-gpu.c
@@ -215,7 +215,7 @@ if ( msg0 && msg2 ) {  /* L(:,k) and U(k,:) are not empty. */
 		    size_t C_stream_size = nbrow * num_col_stream * sizeof(doublecomplex);
 
 		    // Sherry: Check dC buffer of *buffer_size* is large enough
-		    assert(nbrow*(st_col+num_col_stream) < buffer_size);
+		    assert(nbrow*(st_col+num_col_stream) <= buffer_size);
 
 		    gpuMemcpyAsync(dB+b_offset, tempu+b_offset, B_stream_size,
 		    		    gpuMemcpyHostToDevice, streams[stream_id]);

diff --git a/SRC/double/dSchCompUdt-gpu.c b/SRC/double/dSchCompUdt-gpu.c
@@ -216,7 +216,7 @@ if ( msg0 && msg2 ) {  /* L(:,k) and U(k,:) are not empty. */
 		    size_t C_stream_size = nbrow * num_col_stream * sizeof(double);
 
 		    // Sherry: Check dC buffer of *buffer_size* is large enough
-		    assert(nbrow*(st_col+num_col_stream) < buffer_size);
+		    assert(nbrow*(st_col+num_col_stream) <= buffer_size);
 
 		    gpuMemcpyAsync(dB+b_offset, tempu+b_offset, B_stream_size,
 		    		    gpuMemcpyHostToDevice, streams[stream_id]);

diff --git a/SRC/single/sSchCompUdt-gpu.c b/SRC/single/sSchCompUdt-gpu.c
@@ -216,7 +216,7 @@ if ( msg0 && msg2 ) {  /* L(:,k) and U(k,:) are not empty. */
 		    size_t C_stream_size = nbrow * num_col_stream * sizeof(float);
 
 		    // Sherry: Check dC buffer of *buffer_size* is large enough
-		    assert(nbrow*(st_col+num_col_stream) < buffer_size);
+		    assert(nbrow*(st_col+num_col_stream) <= buffer_size);
 
 		    gpuMemcpyAsync(dB+b_offset, tempu+b_offset, B_stream_size,
 		    		    gpuMemcpyHostToDevice, streams[stream_id]);