mindspore训练阶段报错：Not find op[Add] in akg

问题描述：

mindspore训练模块

【操作步骤&问题现象】

1、运行train.py，在optimizer报warning

[WARNING] OPTIMIZER(50958,7f844a0d63c0,python3):2022-02-08-07:59:05.008.715 [mindspore/ccsrc/frontend/optimizer/ad/kprim.cc:287] GetBprop] Fail to find bprop function for IsNan. fn: None

[WARNING] OPTIMIZER(50958,7f844a0d63c0,python3):2022-02-08-07:59:05.010.127 [mindspore/ccsrc/frontend/optimizer/ad/kprim.cc:287] GetBprop] Fail to find bprop function for IsNan. fn: None

[WARNING] OPTIMIZER(50958,7f844a0d63c0,python3):2022-02-08-07:59:05.022.053 [mindspore/ccsrc/frontend/optimizer/ad/kprim.cc:287] GetBprop] Fail to find bprop function for IsNan. fn: None

[WARNING] OPTIMIZER(50958,7f844a0d63c0,python3):2022-02-08-07:59:05.022.625 [mindspore/ccsrc/frontend/optimizer/ad/kprim.cc:287] GetBprop] Fail to find bprop function for IsNan. fn: None

[WARNING] OPTIMIZER(50958,7f844a0d63c0,python3):2022-02-08-07:59:05.092.330 [mindspore/ccsrc/frontend/optimizer/ad/kprim.cc:287] GetBprop] Fail to find bprop function for IsNan. fn: None

2、报错

[ERROR] DEVICE(50958,7f844a0d63c0,python3):2022-02-08-07:59:05.740.893 [mindspore/ccsrc/runtime/device/gpu/kernel_info_setter.cc:123] SelectAkgKernel] Not find op[Add] in akg

[EXCEPTION] DEVICE(50958,7f844a0d63c0,python3):2022-02-08-07:59:05.740.956 [mindspore/ccsrc/runtime/device/gpu/kernel_info_setter.cc:355] PrintUnsupportedTypeException] Select GPU kernel op[Add] fail! Incompatible data type!

The supported data types are in[complex64 complex64], out[complex64]; in[complex64 float32], out[complex64]; in[float32 complex64], out[complex64]; in[complex128 complex128], out[complex128]; in[complex128 float64], out[complex128]; in[float64 complex128], out[complex128]; in[float64 float64], out[float64]; in[float32 float32], out[float32]; in[float16 float16], out[float16]; in[int32 int32], out[int32]; in[int64 int64], out[int64]; , but get in [float32 bool ] out [float32 ]

解答：

 

出现上述error的原因是因为算子不支持，float32和bool相加不支持， 需要将bool转化一下成float32。建议使用mindspore.ops.functional.cast转化数据。