When lowering custom kernel, like mosaic kernel to GPU, how to get the target device/stream ids?

[zhangdaleig]

profiler want to show some event in post processing of the custom call result, highly need the device_id, stream_id of the kernel executing on.