From c5186d84d4466c7e8612bc0638cf76168b8b8d8c Mon Sep 17 00:00:00 2001 From: Masahiro Hiramori Date: Thu, 17 Oct 2024 03:15:09 +0900 Subject: [PATCH] [CI] Upgrade pytorch to 2.4.1 (#784) * upgrade pytorch to 2.4.1 * use onnx opset 13 --- .github/workflows/pythonapp.yml | 2 +- hummingbird/ml/_topology.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml index 6d6f1827d..fe0975465 100644 --- a/.github/workflows/pythonapp.yml +++ b/.github/workflows/pythonapp.yml @@ -32,7 +32,7 @@ jobs: python-version: ['3.9', '3.10', '3.11'] env: TVM_VERSION_TAG: v0.17.0 - PYTORCH_VERSION: 2.2.0 + PYTORCH_VERSION: 2.4.1 LLVM_VERSION: 18 steps: diff --git a/hummingbird/ml/_topology.py b/hummingbird/ml/_topology.py index 38448ebbb..e7a8fc6af 100644 --- a/hummingbird/ml/_topology.py +++ b/hummingbird/ml/_topology.py @@ -243,7 +243,7 @@ def convert(topology, backend, test_input, device, extra_config={}): if backend == onnx.__name__: onnx_model_name = output_model_name = None - target_opset = 11 + target_opset = 13 # Set optional configuration options for ONNX if any. if constants.ONNX_OUTPUT_MODEL_NAME in extra_config: