saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-28 20:11:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Enable cast propagation with level one by default. (#8286)

Browse source
This commit is contained in:
satyajandhyala 2021-07-08 14:38:09 -07:00 committed by GitHub
parent f40df30219
commit 84bc20fe9d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
include/onnxruntime/core/optimizer/graph_transformer_config.h
View file

@ -51,7 +51,7 @@ struct GraphTransformerConfiguration {
return (s1 == s2) == false;
}
int level{-1}; /* -1 => no cast propagation,
int level{1}; /* -1 => no cast propagation,
0 => use user specified list of opcodes to allow moving cast operations,
1 => use ORT predefined list of level 1 opcodes in addition to the user specified allow opcodes
2 => use ORT predefined list of level 2 opcodes in addition to the user specified allow opcodes

2
orttraining/orttraining/python/orttraining_pybind_state.cc
View file

@ -71,7 +71,7 @@ struct TrainingParameters {
bool enable_adasum = false;
// transformation
int propagate_cast_ops_level = -1;
int propagate_cast_ops_level = 1;
std::vector<std::string> propagate_cast_ops_allow;
GraphTransformerConfiguration::PropagateCastOpsConfiguration::Strategy propagate_cast_ops_strategy =
GraphTransformerConfiguration::PropagateCastOpsConfiguration::Strategy::None;