Skip to content

Strongly-typed static code generation for IntelliSense support #162

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 6 commits into from
Nov 29, 2022
Merged

Strongly-typed static code generation for IntelliSense support #162

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
eff8824
Create actual types for TensorType so that they work through type ann…
abock Nov 3, 2022
ccad857
values.Opset: add class to cache key
abock Nov 6, 2022
e35b9f8
Implement strongly typed static code generation of all ops
abock Nov 4, 2022
15198af
Generated opset code
abock Nov 4, 2022
479c906
Generate opset 18 from onnx==1.13.0rc1
abock Nov 28, 2022
48b8029
tensor.py: explicitly downcast opset to Any
abock Nov 27, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions onnxscript/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,10 +26,10 @@
opset16,
opset17,
opset18,
opset_ai_onnx_ml1,
opset_ai_onnx_ml2,
opset_ai_onnx_ml3,
default_opset,
onnxml1,
onnxml2,
onnxml3,
)

from .onnx_types import (
Expand Down Expand Up @@ -99,8 +99,8 @@
"opset16",
"opset17",
"opset18",
"opset_ai_onnx_ml1",
"opset_ai_onnx_ml2",
"opset_ai_onnx_ml3",
"default_opset",
"onnxml1",
"onnxml2",
"onnxml3",
]
39 changes: 0 additions & 39 deletions onnxscript/onnx_opset.py
View file
Open in desktop

This file was deleted.

191 changes: 191 additions & 0 deletions onnxscript/onnx_opset/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,191 @@
# --------------------------------------------------------------------------
# ⚠️ WARNING - AUTO-GENERATED CODE - DO NOT EDIT ⚠️
# ⚙️ Generated by 'python -m opgen'
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# --------------------------------------------------------------------------
# flake8: noqa
# mypy: disable-error-code=override
# pylint: disable=W0221,W0222,W0237,W0246,R0901
# --------------------------------------------------------------------------

from onnx.defs import onnx_opset_version

from onnxscript.onnx_opset._impl.opset1 import Opset1
from onnxscript.onnx_opset._impl.opset2 import Opset2
from onnxscript.onnx_opset._impl.opset3 import Opset3
from onnxscript.onnx_opset._impl.opset4 import Opset4
from onnxscript.onnx_opset._impl.opset5 import Opset5
from onnxscript.onnx_opset._impl.opset6 import Opset6
from onnxscript.onnx_opset._impl.opset7 import Opset7
from onnxscript.onnx_opset._impl.opset8 import Opset8
from onnxscript.onnx_opset._impl.opset9 import Opset9
from onnxscript.onnx_opset._impl.opset10 import Opset10
from onnxscript.onnx_opset._impl.opset11 import Opset11
from onnxscript.onnx_opset._impl.opset12 import Opset12
from onnxscript.onnx_opset._impl.opset13 import Opset13
from onnxscript.onnx_opset._impl.opset14 import Opset14
from onnxscript.onnx_opset._impl.opset15 import Opset15
from onnxscript.onnx_opset._impl.opset16 import Opset16
from onnxscript.onnx_opset._impl.opset17 import Opset17
from onnxscript.onnx_opset._impl.opset18 import Opset18
from onnxscript.onnx_opset._impl.opset_ai_onnx_ml1 import Opset_ai_onnx_ml1
from onnxscript.onnx_opset._impl.opset_ai_onnx_ml2 import Opset_ai_onnx_ml2
from onnxscript.onnx_opset._impl.opset_ai_onnx_ml3 import Opset_ai_onnx_ml3
from onnxscript.onnx_opset._impl.opset_ai_onnx_preview_training1 import (
Opset_ai_onnx_preview_training1,
)

__all__ = [
"default_opset",
"all_opsets",
"opset1",
"opset2",
"opset3",
"opset4",
"opset5",
"opset6",
"opset7",
"opset8",
"opset9",
"opset10",
"opset11",
"opset12",
"opset13",
"opset14",
"opset15",
"opset16",
"opset17",
"opset18",
"opset_ai_onnx_ml1",
"opset_ai_onnx_ml2",
"opset_ai_onnx_ml3",
"opset_ai_onnx_preview_training1",
]


if onnx_opset_version() < 14:
raise ImportError(
f"ONNX Script requires ONNX opset >= 14 but {onnx_opset_version()} is detected."
)


opset1 = Opset1()
opset2 = Opset2()
opset3 = Opset3()
opset4 = Opset4()
opset5 = Opset5()
opset6 = Opset6()
opset7 = Opset7()
opset8 = Opset8()
opset9 = Opset9()
opset10 = Opset10()
opset11 = Opset11()
opset12 = Opset12()
opset13 = Opset13()
opset14 = Opset14()
opset15 = Opset15()
opset16 = Opset16()
opset17 = Opset17()
opset18 = Opset18()
opset_ai_onnx_ml1 = Opset_ai_onnx_ml1()
opset_ai_onnx_ml2 = Opset_ai_onnx_ml2()
opset_ai_onnx_ml3 = Opset_ai_onnx_ml3()
opset_ai_onnx_preview_training1 = Opset_ai_onnx_preview_training1()
all_opsets = {
(
"",
1,
): opset1,
(
"",
2,
): opset2,
(
"",
3,
): opset3,
(
"",
4,
): opset4,
(
"",
5,
): opset5,
(
"",
6,
): opset6,
(
"",
7,
): opset7,
(
"",
8,
): opset8,
(
"",
9,
): opset9,
(
"",
10,
): opset10,
(
"",
11,
): opset11,
(
"",
12,
): opset12,
(
"",
13,
): opset13,
(
"",
14,
): opset14,
(
"",
15,
): opset15,
(
"",
16,
): opset16,
(
"",
17,
): opset17,
(
"",
18,
): opset18,
(
"ai.onnx.ml",
1,
): opset_ai_onnx_ml1,
(
"ai.onnx.ml",
2,
): opset_ai_onnx_ml2,
(
"ai.onnx.ml",
3,
): opset_ai_onnx_ml3,
(
"ai.onnx.preview.training",
1,
): opset_ai_onnx_preview_training1,
}
default_opset: Opset14 = all_opsets[
(
"",
onnx_opset_version(),
)
] # type: ignore
Loading