onnxruntime
78a5f049 - [DML] Model corrupter during layernorm fusion and DmlNonZeroOperator crashes (#16918)

Commit
2 years ago
[DML] Model corrupter during layernorm fusion and DmlNonZeroOperator crashes (#16918) [DML] Model corrupter during layernorm fusion and DmlNonZeroOperator crashes Two issues fixed in this PR: 1) Changes to layernom fusion regressed DirectML. This has been disabled for DML to unblock models. 2) DmlNonZero needs to create an operator call that needs to know the number of non-zero elements (size in bytes). Therefore this needs to be allocated during compute, but is being allocated during initialization. This causes the output tensor size to mismatch with the operator's expectations. --------- Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
Author
Parents
Loading