Skip to content

Commit

Permalink
Changed openvino_options to dict
Browse files Browse the repository at this point in the history
  • Loading branch information
suryasidd committed Feb 9, 2024
1 parent 6de206b commit 903d40f
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
2) model = torch.compile(model, backend="openvino")
"""

openvino_options = []
openvino_options = {}

@register_backend
@fake_tensor_unsupported
Expand Down Expand Up @@ -121,7 +121,7 @@ def _call(*args):

def fx_openvino(subgraph, example_inputs, options=None):
try:
if (openvino_options is not None):
if len(openvino_options) != 0:
options = openvino_options
executor_parameters = None
inputs_reversed = False
Expand All @@ -145,7 +145,7 @@ def _call(*args):
example_inputs.reverse()

from torch._subclasses.fake_tensor import FakeTensorMode
decompositions = _get_decompositions(options)
decompositions = _get_decompositions(options)
if (_get_aot_autograd(options)):
decompositions = decompositions + get_aot_decomposition_list()
with FakeTensorMode(allow_non_fake_inputs=True):
Expand Down

0 comments on commit 903d40f

Please sign in to comment.