Thank you! Your code works great and I’m seeing:
CUDA Exception: Warp Illegal Address
The exception was triggered at PC 0x555591e11f40

Thread 27 "python" received signal CUDA_EXCEPTION_14, Warp Illegal Address.
[Switching focus to CUDA kernel 0, grid 19319, block (84,0,0), thread (192,0,0), device 0, sm 0, warp 20, lane 0]
0x0000555591e12040 in void vision::ops::(anonymous namespace)::roi_align_forward_kernel_impl<float>(--Type <RET> for more, q to quit, c to continue without paging--
int, float const*, float, int, int, int, int, int, int, bool, float const*, float*)
   <<<(172,1,1),(512,1,1)>>> ()

Will debug it further, create an issue, and update you here.