fix: using te and fsdp leads to multiple device found error (#1453)

fix: using te and fsdp leads to multiple device found error (#1453) #783