saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-27 22:45:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Eager mode: Fix some python warnings. (#12167)

Browse source
This commit is contained in:
Wil Brady 2022-07-13 20:24:42 -04:00 committed by GitHub
parent 51f8456c4d
commit 5da1e5d36d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
orttraining/orttraining/eager/opgen/opgen.py
View file

@ -4,13 +4,13 @@
# Licensed under the MIT License.
import argparse
import os
import sys
import types
from importlib.machinery import SourceFileLoader
from opgen.generator import ORTGen as ORTGen
from opgen.generator import ORTGen
from opgen.parser import cpp_create_from_file as CPPParser
from opgen.writer import SourceWriter as SourceWriter
from opgen.writer import SourceWriter
parser = argparse.ArgumentParser(description="Generate ORT ATen operations")
parser.add_argument(
@ -23,7 +23,9 @@ parser.add_argument(
)
args = parser.parse_args()
ops_module = SourceFileLoader("opgen.customop", args.ops_module).load_module()
loader = SourceFileLoader("", args.ops_module)
ops_module = types.ModuleType(loader.name)
loader.exec_module(ops_module)
ortgen = ORTGen(
ops_module.ops,

5
orttraining/orttraining/eager/opgen/opgen/atenops.py
View file

@ -1,10 +1,7 @@
from copy import deepcopy
import torch
from opgen.generator import MakeTorchFallback as MakeTorchFallback
from opgen.generator import ONNXOp as ONNXOp
from opgen.generator import ORTGen as ORTGen
from opgen.generator import SignatureOnly as SignatureOnly
from opgen.generator import MakeTorchFallback, ONNXOp, SignatureOnly
from opgen.onnxops import *
from packaging import version

10
orttraining/orttraining/eager/opgen/opgen/custom_ops.py
View file

@ -1,12 +1,4 @@
from copy import deepcopy
from opgen.generator import AttrType
from opgen.generator import MakeTorchFallback as MakeTorchFallback
from opgen.generator import ONNXAttr
from opgen.generator import ONNXOp as ONNXOp
from opgen.generator import ORTGen as ORTGen
from opgen.generator import SignatureOnly as SignatureOnly
from opgen.onnxops import *
from opgen.onnxops import BatchNormalization, Gemm
ops = {
"gemm": Gemm("A", "B", "C", "alpha", "beta", "transA", "transB"),