From 51d1afd7c43d7d23a3526fd48082669a75d0f33c Mon Sep 17 00:00:00 2001 From: Dang Qingqing Date: Tue, 18 Sep 2018 14:40:18 +0800 Subject: [PATCH] Update code and fix install --- paddle/fluid/API.spec | 4 ++++ python/paddle/fluid/contrib/__init__.py | 1 + python/paddle/fluid/contrib/quantize/__init__.py | 1 + python/setup.py.in | 1 + 4 files changed, 7 insertions(+) diff --git a/paddle/fluid/API.spec b/paddle/fluid/API.spec index e362d34864..45ac698da4 100644 --- a/paddle/fluid/API.spec +++ b/paddle/fluid/API.spec @@ -342,6 +342,10 @@ paddle.fluid.contrib.BeamSearchDecoder.early_stop ArgSpec(args=['self'], varargs paddle.fluid.contrib.BeamSearchDecoder.read_array ArgSpec(args=['self', 'init', 'is_ids', 'is_scores'], varargs=None, keywords=None, defaults=(False, False)) paddle.fluid.contrib.BeamSearchDecoder.update_array ArgSpec(args=['self', 'array', 'value'], varargs=None, keywords=None, defaults=None) paddle.fluid.contrib.memory_usage ArgSpec(args=['program', 'batch_size'], varargs=None, keywords=None, defaults=None) +paddle.fluid.contrib.QuantizeTranspiler.__init__ ArgSpec(args=['self', 'weight_bits', 'activation_bits', 'activation_quantize_type', 'weight_quantize_type', 'window_size'], varargs=None, keywords=None, defaults=(8, 8, 'abs_max', 'abs_max', 10000)) +paddle.fluid.contrib.QuantizeTranspiler.convert_to_int8 ArgSpec(args=['self', 'program', 'place', 'scope'], varargs=None, keywords=None, defaults=(None,)) +paddle.fluid.contrib.QuantizeTranspiler.freeze_program ArgSpec(args=['self', 'program', 'place', 'fuse_bn', 'scope'], varargs=None, keywords=None, defaults=(False, None)) +paddle.fluid.contrib.QuantizeTranspiler.training_transpile ArgSpec(args=['self', 'program', 'startup_program'], varargs=None, keywords=None, defaults=(None, None)) paddle.fluid.transpiler.DistributeTranspiler.__init__ ArgSpec(args=['self', 'config'], varargs=None, keywords=None, defaults=(None,)) paddle.fluid.transpiler.DistributeTranspiler.get_pserver_program ArgSpec(args=['self', 'endpoint'], varargs=None, keywords=None, defaults=None) paddle.fluid.transpiler.DistributeTranspiler.get_pserver_programs ArgSpec(args=['self', 'endpoint'], varargs=None, keywords=None, defaults=None) diff --git a/python/paddle/fluid/contrib/__init__.py b/python/paddle/fluid/contrib/__init__.py index 2d78af05ec..ea9725d1df 100644 --- a/python/paddle/fluid/contrib/__init__.py +++ b/python/paddle/fluid/contrib/__init__.py @@ -19,6 +19,7 @@ from .decoder import * from . import memory_usage_calc from .memory_usage_calc import * from . import quantize +from .quantize import * __all__ = decoder.__all__ + memory_usage_calc.__all__ __all__ += quantize.__all__ diff --git a/python/paddle/fluid/contrib/quantize/__init__.py b/python/paddle/fluid/contrib/quantize/__init__.py index 4e1f3d8dc4..14c208d0e7 100644 --- a/python/paddle/fluid/contrib/quantize/__init__.py +++ b/python/paddle/fluid/contrib/quantize/__init__.py @@ -15,5 +15,6 @@ from __future__ import print_function from . import quantize_transpiler +from .quantize_transpiler import * __all__ = quantize_transpiler.__all__ diff --git a/python/setup.py.in b/python/setup.py.in index 786c9f2e39..b376be0ea3 100644 --- a/python/setup.py.in +++ b/python/setup.py.in @@ -106,6 +106,7 @@ packages=['paddle', 'paddle.fluid.layers', 'paddle.fluid.contrib', 'paddle.fluid.contrib.decoder', + 'paddle.fluid.contrib.quantize', 'paddle.fluid.transpiler', 'paddle.fluid.transpiler.details'] -- GitLab