chck size

8f302c12 · Zoltan Karsa · 22e6404c · 8f302c12
Commit 8f302c12 authored 2 years ago by Zoltan Karsa
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

genax.py
+4 -0

No files found.
--- a/genax.py
+++ b/genax.py
@@ -27,6 +27,8 @@ __global__
 void parosit(const double* x1, const double* x2, double* a, double* b, const int m, const double PI) {
    int tid = blockDim.x * blockIdx.x + threadIdx.x;
    float alpha = x1[tid];
+    if (m*m <= tid*m+m-1)
+        return;
    for (int i = 0; i < m; i++) {
        float betha = x2[i];
        if ((alpha + betha) < PI && betha >= alpha && alpha > 0.0) {
@@ -46,6 +48,8 @@ __global__
 void parosit2(const double* x1, const double* x2, double* a, double* b, const int m, const double PI) {
    int tid = blockDim.x * blockIdx.x + threadIdx.x;
    float alpha = x1[tid];
+    if (m*m <= tid*m+m-1)
+        return;
    for (int i = 0; i < m; i++) {
        float betha = x2[i];
        if ((alpha + betha) < PI && alpha > 0.0) {