Properly include declaration of dispatch in file that registers it. (#30311)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30311
multinomial_stub must be in scope to register against it. Somehow,
this works today, but when I split torch_cpu and torch_cuda it
doesn't.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Test Plan: Imported from OSS
Differential Revision: D18762358
Pulled By: ezyang
fbshipit-source-id: ef9c111292cd02d816af1c94c8bbaadabffaabe5