mirror of https://github.com/llvm/torch-mlir
12 lines
445 B
Python
12 lines
445 B
Python
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
# See https://llvm.org/LICENSE.txt for license information.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
# Also available under a BSD-style license. See LICENSE.
|
|
|
|
from packaging import version
|
|
import torch
|
|
|
|
def torch_version_for_comparison():
|
|
# Ignore +cpu, +cu117m, etc. in comparisons
|
|
return version.parse(torch.__version__.split("+", 1)[0])
|