您好:
csrc/aten/util/common.h内定义了CHECK_MLU_DEVICE() 宏。
算子分发逻辑相对复杂,可通过查看/torch/src/catch/torch_mlu/tools/mlu_functions.yaml 中各个算子的derived_type来判断是否分发到了mlu
请登录后评论