Fix tensor type mismatch: ensure int types for CUDA operations e4c4055 Julien Blanchon commited on Sep 15