Enable distributed unit tests in https://github.com/pytorch/helion/pull/375 once we have multi-card CI.