Skip to content

Commit

Permalink
ci: update compiler config name
Browse files Browse the repository at this point in the history
  • Loading branch information
furiosamg committed May 26, 2023
1 parent 76fc438 commit 7e645fa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions furiosa/models/data/enf_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from furiosa.tools.compiler.api import VersionInfo, compile

QUANTIZER_CONFIG = {"with_quantize": False}
COMPILER_CONFIG = {"tabulate_dequantize": True}
COMPILER_CONFIG = {"lower_tabulated_dequantize": True}
TARGET_NPU = "warboy-2pe"
COMPILED_SUFFIX = "_warboy_2pe"
MAX_WORKER_PROCESSES = 8
Expand Down Expand Up @@ -70,7 +70,7 @@ def quantize_and_compile_model(arg: Tuple[int, Path]):

with open('/dev/null', 'w') as devnull:
# Redirect C lib's stderr to /dev/null
os.dup2(devnull.fileno(), 2)
# os.dup2(devnull.fileno(), 2)

# Compile and write to file
enf = compile(bytes(dfg), target_npu=TARGET_NPU)
Expand Down

0 comments on commit 7e645fa

Please sign in to comment.