Closed fpzh2011 closed 1 month ago
libai/evaluation/evaluator.py 中调用 tensor_to_rank0 时,没有传递 device 参数,这样只能是默认的 cuda,不支持 npu/xpu 等设备。 对于 global tensor,可以用 placement.type 作为 device 参数值。
libai/evaluation/evaluator.py 中调用 tensor_to_rank0 时,没有传递 device 参数,这样只能是默认的 cuda,不支持 npu/xpu 等设备。 对于 global tensor,可以用 placement.type 作为 device 参数值。