Skip to content
Merged
Changes from all commits
Commits
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
16 changes: 10 additions & 6 deletions backends/samsung/serialization/compile_options.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,8 @@
from dataclasses import dataclass
from enum import IntEnum, unique

import pkg_resources
from importlib.resources import files

from executorch.exir._serialize._dataclass import _DataclassEncoder
from executorch.exir._serialize._flatbuffer import _flatc_compile
from executorch.exir.backend.backend_details import CompileSpec
Expand All @@ -36,12 +37,15 @@ def gen_samsung_backend_compile_spec_core(options: EnnExecuTorchOptions) -> Comp
with tempfile.TemporaryDirectory() as d:
# schema
schema_path = os.path.join(d, "{}.fbs".format(COMPILE_OPTION_SCHEMA_NAME))

schema_content = (
files(__package__)
.joinpath(f"{COMPILE_OPTION_SCHEMA_NAME}.fbs")
.read_bytes()
)

with open(schema_path, "wb") as schema_file:
schema_file.write(
pkg_resources.resource_string(
__name__, "{}.fbs".format(COMPILE_OPTION_SCHEMA_NAME)
)
)
schema_file.write(schema_content)
# dump json
json_path = os.path.join(d, "{}.json".format(COMPILE_OPTION_SCHEMA_NAME))
enn_options_json = json.dumps(options, cls=_DataclassEncoder, indent=4)
Expand Down
Loading