I am using ATen and this 37 example, from here 18.
I have a kernel that needs one array of floats (for input) and one array of ints (for labels), should I still use AT_DISPATCH_FLOATING_TYPES?


does this line mean only floats will be marshalled to cuda kernel? or is it just specifying the scalar_t?
AT_DISPATCH_FLOATING_TYPES(X.type(), “lltm_forward_cuda” …


Please correct me if I am wrong: The integral types (int, bool, …) I guess need no special treatment


P.S. it looks like both lines 123 and 157 pass “lltm_forward_cuda” as the second argument, that does not seem right if it is for debugging