[ONNX] Enable opset 13 ORT tests#46903
Conversation
💊 CI failures summary and remediationsAs of commit 83272f8 (more details on the Dr. CI page):
🕵️ 3 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages (reran 1 job to discount flakiness):
|
…oof/opset13 # Conflicts: # .jenkins/caffe2/test.sh
BowenBao
left a comment
There was a problem hiding this comment.
Looks good, thanks! Some minor comments
| if torch.cuda.is_available(): | ||
| torch.cuda.manual_seed_all(0) | ||
| np.random.seed(seed=0) | ||
| os.environ['ALLOW_RELEASED_ONNX_OPSET_ONLY'] = '0' |
There was a problem hiding this comment.
curious why we need this?
There was a problem hiding this comment.
This is to enable opset 13 in ORT before official release.
|
|
||
| # opset 13 tests, with keep_initializers_as_inputs=False for | ||
| # IR version 4 style export. | ||
| TestONNXRuntime_opset13_IRv4 = type(str("TestONNXRuntime_opset13_IRv4"), |
There was a problem hiding this comment.
actually not sure if there is still the need of separate tests on IR versions, since sequence proto requires IR version > 4 anyways ...
|
|
||
| _default_onnx_opset_version = 9 | ||
| _onnx_master_opset = 10 | ||
| _onnx_main_opset = 13 |
| import torch.onnx.symbolic_helper as sym_help | ||
| from torch.onnx.symbolic_helper import parse_args, _unimplemented | ||
|
|
||
| block_listed_operators = ['embedding_bag'] |
There was a problem hiding this comment.
is it due to loop that embedding_bag not supported?
03f815f to
04f5505
Compare
Enable export of opset 13 ops.