• W
    [Cherry-Pick] Fix the segfault when using to_tensor in PyLayer. (#33303) (#33518) · 0079e0b1
    WeiXin 提交于
    修复pylayer 返回to_tensor时触发段错误的bug。
    原因:
    
    如果在Python端修改了stop_gradient属性,c++ 端InnerSetOverridedStopGradient 无法修改stop_gradient属性,在c++端调用SetOverridedStopGradient修改stop_gradient属性。
    to_tensor产生的tensor的grad var的DataType为默认值(-1),在backward的过程中grad var的DataType不能为默认值(-1),因此在调用ForwardDataType设置grad var的DataType。
    
    原始PR:#33303
    0079e0b1
test_pylayer_op.py 16.4 KB