[Disco] Expose functions to query the per-worker device/rank#16639

masahi merged 1 commit intoapache:mainapache/tvm:mainfrom

Lunderberg:disco_test_callback_for_param_loadingLunderberg/tvm:disco_test_callback_for_param_loadingCopy head branch name to clipboard