Revert "[jiterator] Reduce templating in jitted_gpu_kernel_impl (#80103)"
This reverts commit df665b1a9d7506ee147524d02fc4639a3b27c56f.
Reverted https://github.com/pytorch/pytorch/pull/80103 on behalf of https://github.com/mehtanirav due to internal breakage
