Open
Description
The snippet
xla/torch_xla/experimental/scan.py
Lines 217 to 226 in 00fac78
require_grads=True
on all carry inputs and that won't work if one of the carry is a LongTensor
.
The most obvious example is that if one of the input is an integer, then it can't possibly have gradients.