Skip to content

Commit

Permalink
bump version to 0.0.3 (#156)
Browse files Browse the repository at this point in the history
* bump version

* bump version in setup.cfg
  • Loading branch information
interesaaat authored Jun 18, 2020
1 parent 79c4d75 commit 6fee72c
Show file tree
Hide file tree
Showing 5 changed files with 17 additions and 16 deletions.
2 changes: 1 addition & 1 deletion doc/html/hummingbird/index.html
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ <h1 class="title">Package <code>hummingbird</code></h1>
Entrypoint for Hummingbird modules.
&#34;&#34;&#34;

__version__ = &#34;0.0.2&#34;
__version__ = &#34;0.0.3&#34;
__author__ = &#34;Microsoft&#34;
__producer__ = &#34;hummingbird&#34;
__producer_version__ = __version__
Expand Down
26 changes: 14 additions & 12 deletions doc/html/hummingbird/ml/ir_converters/linked_node.html
Original file line number Diff line number Diff line change
Expand Up @@ -114,12 +114,13 @@ <h1 class="title">Module <code>hummingbird.ml.ir_converters.linked_node</code></
# Convert the model from PyTorch to ONNX.
if constants.N_FEATURES not in extra_config:
extra_config[constants.ONNX_INPUTS] = all_tensors
# Early version of pytorch have a bug with exporting gemm into ONNX.
trimmed_version = torch.__version__.split(&#34;+&#34;)[0] # For Win pytorch has a +cpu or +gpu postfix
torch_version = LooseVersion(trimmed_version)
gemm_min = LooseVersion(&#34;1.5.0&#34;)
if torch_version &lt;= gemm_min:
extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
# Pytorch has a bug with exporting gemm into ONNX.
# For the moment only tree_trav is enabled
# trimmed_version = torch.__version__.split(&#34;+&#34;)[0] # For Win pytorch has a +cpu or +gpu postfix
# torch_version = LooseVersion(trimmed_version)
# gemm_min = LooseVersion(&#34;1.5.1&#34;)
# if torch_version &lt;= gemm_min:
extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
pytorch_model = converter(node_, extra_config=extra_config)

# Generate the inputs for the tracing.
Expand Down Expand Up @@ -444,12 +445,13 @@ <h2 id="returns">Returns</h2>
# Convert the model from PyTorch to ONNX.
if constants.N_FEATURES not in extra_config:
extra_config[constants.ONNX_INPUTS] = all_tensors
# Early version of pytorch have a bug with exporting gemm into ONNX.
trimmed_version = torch.__version__.split(&#34;+&#34;)[0] # For Win pytorch has a +cpu or +gpu postfix
torch_version = LooseVersion(trimmed_version)
gemm_min = LooseVersion(&#34;1.5.0&#34;)
if torch_version &lt;= gemm_min:
extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
# Pytorch has a bug with exporting gemm into ONNX.
# For the moment only tree_trav is enabled
# trimmed_version = torch.__version__.split(&#34;+&#34;)[0] # For Win pytorch has a +cpu or +gpu postfix
# torch_version = LooseVersion(trimmed_version)
# gemm_min = LooseVersion(&#34;1.5.1&#34;)
# if torch_version &lt;= gemm_min:
extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
pytorch_model = converter(node_, extra_config=extra_config)

# Generate the inputs for the tracing.
Expand Down
2 changes: 1 addition & 1 deletion hummingbird/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
Entrypoint for Hummingbird modules.
"""

__version__ = "0.0.2"
__version__ = "0.0.3"
__author__ = "Microsoft"
__producer__ = "hummingbird"
__producer_version__ = __version__
Expand Down
2 changes: 1 addition & 1 deletion setup.cfg
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
[bumpversion]
current_version = 0.0.2
current_version = 0.0.3

[bdist_wheel]
universal = 1
Expand Down
1 change: 0 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
assert packages

# read version from the package file.
version_str = "0.0.2"
with (open(os.path.join(this, "hummingbird/__init__.py"), "r")) as f:
line = [_ for _ in [_.strip("\r\n ") for _ in f.readlines()] if _.startswith("__version__")]
if len(line) > 0:
Expand Down

0 comments on commit 6fee72c

Please sign in to comment.