mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
02e35ce17b
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67803 * Addresses comments from #63589 [ONNX] remove torch::onnx::PRODUCER_VERSION (#67107) Use constants from version.h instead. This simplifies things since we no longer have to update PRODUCER_VERSION for each release. Also add TORCH_VERSION to version.h so that a string is available for this purpose. [ONNX] Set `ir_version` based on opset_version. (#67128) This increases the odds that the exported ONNX model will be usable. Before this change, we were setting the IR version to a value which may be higher than what the model consumer supports. Also some minor clean-up in the test code: * Fix string replacement. * Use a temporary file so as to not leave files around in the test current working directory. Test Plan: Imported from OSS Reviewed By: msaroufim Differential Revision: D32181306 Pulled By: malfet fbshipit-source-id: 02f136d34ef8f664ade0bc1985a584f0e8c2b663 Co-authored-by: BowenBao <bowbao@microsoft.com> Co-authored-by: Gary Miguel <garymiguel@microsoft.com> Co-authored-by: Nikita Shulga <nshulga@fb.com>
30 lines
642 B
ReStructuredText
30 lines
642 B
ReStructuredText
Library Versioning
|
|
==================
|
|
|
|
We provide version number macros for identifying the version of LibTorch in use.
|
|
Example usage:
|
|
|
|
.. code-block:: cpp
|
|
|
|
#include <torch/torch.h>
|
|
#include <iostream>
|
|
|
|
int main() {
|
|
std::cout << "PyTorch version from parts: "
|
|
<< TORCH_VERSION_MAJOR << "."
|
|
<< TORCH_VERSION_MINOR << "."
|
|
<< TORCH_VERSION_PATCH << std::endl;
|
|
std::cout << "PyTorch version: " << TORCH_VERSION << std::endl;
|
|
}
|
|
|
|
This will output something like:
|
|
|
|
.. code-block:: text
|
|
|
|
PyTorch version from parts: 1.8.0
|
|
PyTorch version: 1.8.0
|
|
|
|
.. note::
|
|
|
|
These macros are only available in PyTorch >= 1.8.0.
|