From 04dd6d5d0e5fd205156a23609bfaac443746a669 Mon Sep 17 00:00:00 2001 From: Jacob Szwejbka Date: Tue, 15 Aug 2023 09:47:21 -0700 Subject: [PATCH] delete py_bin for quant. Add message to indicate succesful finish (#61) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/61 instead of using buck for this we should do 'python3 -m examples.quantization.example --model_name mv2' Reviewed By: larryliu0820 Differential Revision: D48325202 fbshipit-source-id: 358d0a593d8893e44dc30d1dd2e0950fcf238035 --- examples/quantization/TARGETS | 4 ++-- examples/quantization/example.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/quantization/TARGETS b/examples/quantization/TARGETS index 231c56567d4..7d8471ac576 100644 --- a/examples/quantization/TARGETS +++ b/examples/quantization/TARGETS @@ -1,6 +1,6 @@ -load("@fbcode_macros//build_defs:python_binary.bzl", "python_binary") +load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") -python_binary( +runtime.python_binary( name = "example", main_src = "example.py", deps = [ diff --git a/examples/quantization/example.py b/examples/quantization/example.py index 009ccd6a486..ef6695443e3 100644 --- a/examples/quantization/example.py +++ b/examples/quantization/example.py @@ -133,3 +133,4 @@ def verify_xnnpack_quantizer_matching_fx_quant_model(model_name, model, example_ ) quantize(args.model_name, model, example_inputs) + print("finished")