Description

arrayfire::locate fails for input large arrays. I only tested this on the rust version of arrayfire. No sure if it appears in the C++ version.

Reproducible Code and/or Steps

For a large array of A, arrayfire locate fails

arrayfire::set_seed(100);
let A_dims = arrayfire::Dim4::new(&[107500,1,32000,1]);
let mut A = arrayfire::randu::<bool>(A_dims);

let result = arrayfire::locate(&A);

println!("  result.dims()[0] {}",result.dims()[0]);
println!("  result.dims()[1] {}",result.dims()[1]);
println!("  result.dims()[2] {}",result.dims()[2]);	

[1]    84261 segmentation fault (core dumped) 

System Information

ArrayFire v3.9.0 (CUDA, 64-bit Linux, build 715e21f)
Platform: CUDA Runtime 12.0, Driver: 525.89.02
[0] NVIDIA GeForce RTX 3090, 24260 MB, CUDA Compute 8.6
Info String:
ArrayFire v3.9.0 (CUDA, 64-bit Linux, build 715e21f)
Platform: CUDA Runtime 12.0, Driver: 525.89.02
[0] NVIDIA GeForce RTX 3090, 24260 MB, CUDA Compute 8.6
Arrayfire version: (3, 9, 0)
Name: NVIDIA_GeForce_RTX_3090
Platform: CUDA
Toolkit: v12.0
Compute: 8.6
Revision: 715e21f

Checklist

• Using the latest available ArrayFire release
• GPU drivers are up to date