chakra
chakra copied to clipboard
what do I specify for the argument --rank when linking the traces on single node single gpu?
I am testing my chakra environment following the steps in wiki. when I link the pytorch trace and kineto trace, with the cmd "chakra_trace_link --chakra-host-trace pytorch_et.json --chakra-device-trace kineto_trace.json --output-file /path/to/chakra_host_device_trace.json ", I got "chakra_trace_link: error: the following arguments are required: --rank". But my host trace and kineto trace are collected in my node with single GPU and without DDP, then what I can provide for the rank argument? look forward for any advice.
Hey @811454567, you can use "--rank 0" for single node traces. Let me know if you need anything else!