Check bounds

ce550441 · Paul · 7339860e · ce550441
Commit ce550441 authored Jul 01, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

src/targets/gpu/kernels/include/migraphx/kernels/index.hpp src/targets/gpu/kernels/include/migraphx/kernels/index.hpp +5 -1

No files found.
--- a/src/targets/gpu/kernels/include/migraphx/kernels/index.hpp
+++ b/src/targets/gpu/kernels/include/migraphx/kernels/index.hpp
@@ -80,7 +80,11 @@ struct index
            }
            else
            {
-                repeat(max_stride_iterations(n, stride), [&](auto i) { f(start + stride * i); });
+                repeat(max_stride_iterations(n, stride), [&](auto k) { 
+                    auto i = start + stride * k;
+                    if (i < n)
+                        f(i); 
+                });
            }
        }
        else