This is an automated email from the ASF dual-hosted git repository.
mshr pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 92e2bba08e [CI] Upgrade CI image to `20241105-030952-3e386fd3` (#17451)
92e2bba08e is described below
commit 92e2bba08e8a8915d5875cd49e522dbdf74c9bca
Author: Masahiro Hiramori <[email protected]>
AuthorDate: Tue Nov 26 10:38:25 2024 +0900
[CI] Upgrade CI image to `20241105-030952-3e386fd3` (#17451)
* use `20241105-030952-3e386fd3` for ci
* fix missing `language` kwarg in save_rst_example
* disable invalid-name check by pylint
* disable caffe frontend tests
* disable mxnet frontend tests
* fix stablehlo importer
* enable some stablehlo tests
* skip tests because mxnet raise AttributeError
* use stable sort for `np.argsort`
* fix `TypeError: arrays to stack must be passed as a "sequence" type such
as list or tuple.`
* disable tests because of shape error
* remove `get_html_theme_path` because it's deprecated
* ignore warnings from sphinx
* disable oneflow frontend tests
* remove oneflow tutorial
* remove debug print
---
ci/jenkins/docker-images.ini | 20 +--
ci/jenkins/unity_jenkinsfile.groovy | 8 +-
docs/conf.py | 11 +-
gallery/how_to/compile_models/from_oneflow.py | 182 ---------------------
python/tvm/contrib/sparse.py | 2 +-
python/tvm/relax/expr.py | 2 +-
.../frontend/stablehlo/stablehlo_translator.py | 8 +-
python/tvm/tir/schedule/schedule.py | 3 +-
tests/python/codegen/test_target_codegen_vulkan.py | 2 +-
.../codegen/test_target_texture_codegen_opencl.py | 2 +
tests/python/contrib/test_cublas.py | 14 +-
tests/python/frontend/coreml/test_forward.py | 9 +-
tests/python/relax/test_frontend_stablehlo.py | 24 ---
tests/scripts/task_python_docs.sh | 2 +
tests/scripts/task_python_frontend.sh | 6 -
tests/scripts/task_python_frontend_cpu.sh | 3 -
16 files changed, 47 insertions(+), 251 deletions(-)
diff --git a/ci/jenkins/docker-images.ini b/ci/jenkins/docker-images.ini
index 175917f887..6d42008c43 100644
--- a/ci/jenkins/docker-images.ini
+++ b/ci/jenkins/docker-images.ini
@@ -17,13 +17,13 @@
# This data file is read during when Jenkins runs job to determine docker
images.
[jenkins]
-ci_arm: tlcpack/ci-arm:20240917-153130-9f281758
-ci_cortexm: tlcpack/ci-cortexm:20240917-153130-9f281758
-ci_cpu: tlcpack/ci_cpu:20240917-153130-9f281758
-ci_gpu: tlcpack/ci-gpu:20240917-153130-9f281758
-ci_hexagon: tlcpack/ci-hexagon:20240917-153130-9f281758
-ci_i386: tlcpack/ci-i386:20240917-153130-9f281758
-ci_lint: tlcpack/ci-lint:20240917-153130-9f281758
-ci_minimal: tlcpack/ci-minimal:20240917-153130-9f281758
-ci_riscv: tlcpack/ci-riscv:20240917-153130-9f281758
-ci_wasm: tlcpack/ci-wasm:20240917-153130-9f281758
+ci_arm: tlcpack/ci-arm:20241119-020227-6fc0598c
+ci_cortexm: tlcpack/ci-cortexm:20241119-020227-6fc0598c
+ci_cpu: tlcpack/ci_cpu:20241119-020227-6fc0598c
+ci_gpu: tlcpack/ci-gpu:20241119-020227-6fc0598c
+ci_hexagon: tlcpack/ci-hexagon:20241119-020227-6fc0598c
+ci_i386: tlcpack/ci-i386:20241119-020227-6fc0598c
+ci_lint: tlcpack/ci-lint:20241119-020227-6fc0598c
+ci_minimal: tlcpack/ci-minimal:20241119-020227-6fc0598c
+ci_riscv: tlcpack/ci-riscv:20241119-020227-6fc0598c
+ci_wasm: tlcpack/ci-wasm:20241119-020227-6fc0598c
diff --git a/ci/jenkins/unity_jenkinsfile.groovy
b/ci/jenkins/unity_jenkinsfile.groovy
index 3e6213ff26..d6a5d46f6f 100755
--- a/ci/jenkins/unity_jenkinsfile.groovy
+++ b/ci/jenkins/unity_jenkinsfile.groovy
@@ -30,14 +30,14 @@
import org.jenkinsci.plugins.pipeline.modeldefinition.Utils
// NOTE: these lines are scanned by docker/dev_common.sh. Please update the
regex as needed. -->
-ci_lint = 'tlcpack/ci_lint:20240917-153130-9f281758'
-ci_gpu = 'tlcpack/ci_gpu:20240917-153130-9f281758'
-ci_cpu = 'tlcpack/ci_cpu:20240917-153130-9f281758'
+ci_lint = 'tlcpack/ci_lint:20241119-020227-6fc0598c'
+ci_gpu = 'tlcpack/ci_gpu:20241119-020227-6fc0598c'
+ci_cpu = 'tlcpack/ci_cpu:20241119-020227-6fc0598c'
ci_wasm = 'tlcpack/ci-wasm:v0.72'
ci_i386 = 'tlcpack/ci-i386:v0.75'
ci_qemu = 'tlcpack/ci-qemu:v0.11'
ci_arm = 'tlcpack/ci-arm:v0.08'
-ci_hexagon = 'tlcpack/ci_hexagon:20240917-153130-9f281758'
+ci_hexagon = 'tlcpack/ci_hexagon:20241119-020227-6fc0598c'
// <--- End of regex-scanned config.
// Parameters to allow overriding (in Jenkins UI), the images
diff --git a/docs/conf.py b/docs/conf.py
index c858e9c450..d686a2d72d 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -159,7 +159,9 @@ BUTTON = (
@monkey_patch("sphinx_gallery.gen_rst", "save_rst_example")
-def save_rst_example(example_rst, example_file, time_elapsed, memory_used,
gallery_conf, real_func):
+def save_rst_example(
+ example_rst, example_file, time_elapsed, memory_used, gallery_conf,
language, real_func
+):
"""Monkey-patch save_rst_example to include the "Open in Colab" button."""
# The url is the md5 hash of the notebook path.
@@ -178,7 +180,9 @@ def save_rst_example(example_rst, example_file,
time_elapsed, memory_used, galle
python_file=example_fname, ref_name=ref_fname, colab_url=colab_url,
button_svg=BUTTON
)
with patch("sphinx_gallery.gen_rst.EXAMPLE_HEADER", new_header):
- real_func(example_rst, example_file, time_elapsed, memory_used,
gallery_conf)
+ real_func(
+ example_rst, example_file, time_elapsed, memory_used,
gallery_conf, language=language
+ )
INCLUDE_DIRECTIVE_RE = re.compile(r"^([ \t]*)\.\. include::\s*(.+)\n",
flags=re.M)
@@ -365,10 +369,7 @@ html_theme = os.environ.get("TVM_THEME", "rtd")
on_rtd = os.environ.get("READTHEDOCS", None) == "True"
# only import rtd theme and set it if want to build docs locally
if not on_rtd and html_theme == "rtd":
- import sphinx_rtd_theme
-
html_theme = "sphinx_rtd_theme"
- html_theme_path = [sphinx_rtd_theme.get_html_theme_path()]
# Add any paths that contain custom static files (such as style sheets) here,
# relative to this directory. They are copied after the builtin static files,
diff --git a/gallery/how_to/compile_models/from_oneflow.py
b/gallery/how_to/compile_models/from_oneflow.py
deleted file mode 100644
index 64f659316b..0000000000
--- a/gallery/how_to/compile_models/from_oneflow.py
+++ /dev/null
@@ -1,182 +0,0 @@
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-"""
-Compile OneFlow Models
-======================
-**Author**: `Xiaoyu Zhang <https://github.com/BBuf/>`_
-
-This article is an introductory tutorial to deploy OneFlow models with Relay.
-
-For us to begin with, OneFlow package should be installed.
-
-A quick solution is to install via pip
-
-.. code-block:: bash
-
- %%shell
- pip install flowvision==0.1.0
- pip install -f https://release.oneflow.info oneflow==0.7.0+cpu
-
-or please refer to official site:
-https://github.com/Oneflow-Inc/oneflow
-
-Currently, TVM supports OneFlow 0.7.0. Other versions may be unstable.
-"""
-
-# sphinx_gallery_start_ignore
-# sphinx_gallery_requires_cuda = True
-# sphinx_gallery_end_ignore
-import os, math
-from matplotlib import pyplot as plt
-import numpy as np
-from PIL import Image
-
-# oneflow imports
-import flowvision
-import oneflow as flow
-import oneflow.nn as nn
-
-import tvm
-from tvm import relay
-from tvm.contrib.download import download_testdata
-
-######################################################################
-# Load a pretrained OneFlow model and save model
-# ----------------------------------------------
-model_name = "resnet18"
-model = getattr(flowvision.models, model_name)(pretrained=True)
-model = model.eval()
-
-model_dir = "resnet18_model"
-if not os.path.exists(model_dir):
- flow.save(model.state_dict(), model_dir)
-
-######################################################################
-# Load a test image
-# -----------------
-# Classic cat example!
-from PIL import Image
-
-img_url = "https://github.com/dmlc/mxnet.js/blob/main/data/cat.png?raw=true";
-img_path = download_testdata(img_url, "cat.png", module="data")
-img = Image.open(img_path).resize((224, 224))
-
-# Preprocess the image and convert to tensor
-from flowvision import transforms
-
-my_preprocess = transforms.Compose(
- [
- transforms.Resize(256),
- transforms.CenterCrop(224),
- transforms.ToTensor(),
- transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224,
0.225]),
- ]
-)
-img = my_preprocess(img)
-img = np.expand_dims(img.numpy(), 0)
-
-######################################################################
-# Import the graph to Relay
-# -------------------------
-# Convert OneFlow graph to Relay graph. The input name can be arbitrary.
-class Graph(flow.nn.Graph):
- def __init__(self, module):
- super().__init__()
- self.m = module
-
- def build(self, x):
- out = self.m(x)
- return out
-
-
-graph = Graph(model)
-_ = graph._compile(flow.randn(1, 3, 224, 224))
-
-mod, params = relay.frontend.from_oneflow(graph, model_dir)
-
-######################################################################
-# Relay Build
-# -----------
-# Compile the graph to llvm target with given input specification.
-target = tvm.target.Target("llvm", host="llvm")
-dev = tvm.cpu(0)
-with tvm.transform.PassContext(opt_level=3):
- lib = relay.build(mod, target=target, params=params)
-
-######################################################################
-# Execute the portable graph on TVM
-# ---------------------------------
-# Now we can try deploying the compiled model on target.
-target = "cuda"
-with tvm.transform.PassContext(opt_level=10):
- intrp = relay.build_module.create_executor("graph", mod, tvm.cuda(0),
target)
-
-print(type(img))
-print(img.shape)
-tvm_output = intrp.evaluate()(tvm.nd.array(img.astype("float32")), **params)
-
-#####################################################################
-# Look up synset name
-# -------------------
-# Look up prediction top 1 index in 1000 class synset.
-synset_url = "".join(
- [
- "https://raw.githubusercontent.com/Cadene/";,
- "pretrained-models.pytorch/master/data/",
- "imagenet_synsets.txt",
- ]
-)
-synset_name = "imagenet_synsets.txt"
-synset_path = download_testdata(synset_url, synset_name, module="data")
-with open(synset_path) as f:
- synsets = f.readlines()
-
-synsets = [x.strip() for x in synsets]
-splits = [line.split(" ") for line in synsets]
-key_to_classname = {spl[0]: " ".join(spl[1:]) for spl in splits}
-
-class_url = "".join(
- [
- "https://raw.githubusercontent.com/Cadene/";,
- "pretrained-models.pytorch/master/data/",
- "imagenet_classes.txt",
- ]
-)
-class_name = "imagenet_classes.txt"
-class_path = download_testdata(class_url, class_name, module="data")
-with open(class_path) as f:
- class_id_to_key = f.readlines()
-
-class_id_to_key = [x.strip() for x in class_id_to_key]
-
-# Get top-1 result for TVM
-top1_tvm = np.argmax(tvm_output.numpy()[0])
-tvm_class_key = class_id_to_key[top1_tvm]
-
-# Convert input to OneFlow variable and get OneFlow result for comparison
-with flow.no_grad():
- torch_img = flow.from_numpy(img)
- output = model(torch_img)
-
- # Get top-1 result for OneFlow
- top_oneflow = np.argmax(output.numpy())
- oneflow_class_key = class_id_to_key[top_oneflow]
-
-print("Relay top-1 id: {}, class name: {}".format(top1_tvm,
key_to_classname[tvm_class_key]))
-print(
- "OneFlow top-1 id: {}, class name: {}".format(top_oneflow,
key_to_classname[oneflow_class_key])
-)
diff --git a/python/tvm/contrib/sparse.py b/python/tvm/contrib/sparse.py
index d561c5cbb1..9f94ff24f9 100644
--- a/python/tvm/contrib/sparse.py
+++ b/python/tvm/contrib/sparse.py
@@ -95,7 +95,7 @@ class CSRNDArray(object):
"""Construct a full matrix and convert it to numpy array."""
full = _np.zeros(self.shape, self.dtype)
ridx = _np.diff(self.indptr.numpy())
- ridx = _np.hstack((_np.ones((v,), itype) * i for i, v in
enumerate(ridx)))
+ ridx = _np.hstack([_np.ones((v,), itype) * i for i, v in
enumerate(ridx)])
full[ridx, self.indices.numpy().astype(itype)] = self.data.numpy()
return full
diff --git a/python/tvm/relax/expr.py b/python/tvm/relax/expr.py
index 522eb11d6d..190df42860 100644
--- a/python/tvm/relax/expr.py
+++ b/python/tvm/relax/expr.py
@@ -39,7 +39,7 @@ from . import _ffi_api
# This feature is not supported until python 3.10:
# https://docs.python.org/3.10/whatsnew/3.10.html#pep-613-typealias
Expr = Union[tvm.ir.RelayExpr]
-Type = Union[tvm.ir.Type]
+Type = Union[tvm.ir.Type] # pylint: disable=invalid-name
GlobalVar = Union[tvm.ir.GlobalVar]
diff --git a/python/tvm/relax/frontend/stablehlo/stablehlo_translator.py
b/python/tvm/relax/frontend/stablehlo/stablehlo_translator.py
index 1ca0856f63..eb8c95b964 100644
--- a/python/tvm/relax/frontend/stablehlo/stablehlo_translator.py
+++ b/python/tvm/relax/frontend/stablehlo/stablehlo_translator.py
@@ -432,14 +432,12 @@ def from_stablehlo(
output : tvm.IRModule
The result IRModule with entry function "main"
"""
- from jaxlib import mlir
- from jaxlib.mlir.dialects import stablehlo
+ from jax._src.interpreters import mlir as jax_mlir
if isinstance(stablehlo_module, str):
# TODO (yongwww): support the serialized bytecode format of StableHLO
# model using stablehlo.deserialize_portable_artifact(ir) if the python
# binding is ready
- with mlir.ir.Context() as context:
- stablehlo.register_dialect(context)
- stablehlo_module = mlir.ir.Module.parse(stablehlo_module)
+ context = jax_mlir.make_ir_context()
+ stablehlo_module = jax_mlir.ir.Module.parse(stablehlo_module, context)
return StableHLOImporter().from_stablehlo(stablehlo_module, input_info)
diff --git a/python/tvm/tir/schedule/schedule.py
b/python/tvm/tir/schedule/schedule.py
index 17c256be35..9e8b4dc34f 100644
--- a/python/tvm/tir/schedule/schedule.py
+++ b/python/tvm/tir/schedule/schedule.py
@@ -61,7 +61,8 @@ class BlockRV(Object):
# It is a workaround for mypy:
https://github.com/python/mypy/issues/7866#issuecomment-549454370
# This feature is not supported until python 3.10:
# https://docs.python.org/3.10/whatsnew/3.10.html#pep-613-typealias
-ExprRV = Union[PrimExpr] # A random variable that evaluates to an integer
+# A random variable that evaluates to an integer
+ExprRV = Union[PrimExpr] # pylint: disable=invalid-name
RAND_VAR_TYPE = Union[ExprRV, BlockRV, LoopRV] # pylint: disable=invalid-name
diff --git a/tests/python/codegen/test_target_codegen_vulkan.py
b/tests/python/codegen/test_target_codegen_vulkan.py
index a8d1719ff2..6973040cb2 100644
--- a/tests/python/codegen/test_target_codegen_vulkan.py
+++ b/tests/python/codegen/test_target_codegen_vulkan.py
@@ -257,7 +257,7 @@ def test_argsort(target, dev):
mod = tvm.IRModule()
mod["main"] = relay.Function([x], relay.argsort(x))
x_np = np.random.randint(0, high=10, size=(10,)).astype(dtype)
- res_np = np.argsort(x_np)
+ res_np = np.argsort(x_np, kind="stable")
check_mod(target, dev, mod, x_np, res_np)
diff --git a/tests/python/codegen/test_target_texture_codegen_opencl.py
b/tests/python/codegen/test_target_texture_codegen_opencl.py
index 5681dcf9e6..213ad8d7ba 100644
--- a/tests/python/codegen/test_target_texture_codegen_opencl.py
+++ b/tests/python/codegen/test_target_texture_codegen_opencl.py
@@ -1385,6 +1385,7 @@ class TestConv2dWCHNcCRSKk(BaseConv2DValidator):
test_func = tvm.testing.parameter(conv2d_1x1_WCHNc_CRSKk)
[email protected]("AttributeError: module 'numpy' has no attribute 'bool'
raised from mxnet")
class TestConv2dNCHWcKCRSk(BaseConv2DValidator):
input_shapes = tvm.testing.parameter(
[(1, 32, 56, 56, 4), (32, 128, 1, 1, 4)], [(1, 32, 112, 112, 4), (32,
128, 3, 3, 4)]
@@ -1392,6 +1393,7 @@ class TestConv2dNCHWcKCRSk(BaseConv2DValidator):
test_func = tvm.testing.parameter(conv2d_NCHWc_KCRSk,
conv2d_NCHWc_KCRSk_fp32_acc)
[email protected]("AttributeError: module 'numpy' has no attribute 'bool'
raised from mxnet")
class TestDepthwiseConv2dNCHWcKCRSk(BaseConv2DValidator):
input_shapes = tvm.testing.parameter([(1, 24, 257, 257, 4), (24, 1, 3, 3,
4)])
test_func = tvm.testing.parameter(depthwise_conv2d_NCHWc_KCRSk_acc32)
diff --git a/tests/python/contrib/test_cublas.py
b/tests/python/contrib/test_cublas.py
index f3e3aa74af..99611bab49 100644
--- a/tests/python/contrib/test_cublas.py
+++ b/tests/python/contrib/test_cublas.py
@@ -82,8 +82,8 @@ def verify_matmul_add_igemm(in_dtype, out_dtype, rtol=1e-5):
b_old = np.random.uniform(0, 128, size=(l, m))
# Transform a to become CUBLASLT_ORDER_COL4_4R2_8C layout
- a_new = np.hstack((a_old.astype(A.dtype), np.zeros([n, L - l])))
- a_new = np.vstack((a_new.astype(A.dtype), np.zeros([N - n, L])))
+ a_new = np.hstack([a_old.astype(A.dtype), np.zeros([n, L - l])])
+ a_new = np.vstack([a_new.astype(A.dtype), np.zeros([N - n, L])])
a_even = np.vsplit(a_new[::2], N / 8)
a_odd = np.vsplit(a_new[1::2], N / 8)
a_new = [None] * (len(a_even) + len(a_odd))
@@ -91,13 +91,17 @@ def verify_matmul_add_igemm(in_dtype, out_dtype, rtol=1e-5):
a_new[1::2] = a_odd
a_new = np.vstack(a_new)
a_new = np.vstack(
- np.vstack(np.vstack(np.hsplit(i, 8)).reshape([4, 32]) for i in
np.vsplit(j, N / 4))
- for j in np.hsplit(a_new, L / 32)
+ [
+ np.vstack(
+ [np.vstack(np.hsplit(i, 8)).reshape([4, 32]) for i in
np.vsplit(j, N / 4)]
+ )
+ for j in np.hsplit(a_new, L / 32)
+ ]
)
a_new = a_new.reshape([N, L])
# Transform b to become CUBLASLT_ORDER_COL32 layout
b_new = np.vstack(
- np.hsplit(np.hstack((b_old.T.astype(B.dtype), np.zeros([m, L -
l]))), L / 32)
+ np.hsplit(np.hstack([b_old.T.astype(B.dtype), np.zeros([m, L -
l])]), L / 32)
)
b_new = b_new.reshape([m, L])
diff --git a/tests/python/frontend/coreml/test_forward.py
b/tests/python/frontend/coreml/test_forward.py
index e381f44734..26ddcba6ef 100644
--- a/tests/python/frontend/coreml/test_forward.py
+++ b/tests/python/frontend/coreml/test_forward.py
@@ -306,7 +306,7 @@ def _verify_average(input_dim1, input_dim2, axis=0):
a_np1 = np.random.uniform(size=input_dim1).astype(dtype)
a_np2 = np.random.uniform(size=input_dim2).astype(dtype)
- b_np = np.mean((a_np1, a_np2), axis=axis)
+ b_np = np.mean((a_np1, a_np2), axis=axis, dtype=float)
inputs = [("input1", datatypes.Array(*input_dim1)), ("input2",
datatypes.Array(*input_dim2))]
output = [("output", datatypes.Array(*b_np.shape))]
@@ -325,8 +325,11 @@ def _verify_average(input_dim1, input_dim2, axis=0):
@tvm.testing.uses_gpu
def test_forward_average():
_verify_average((1, 3, 20, 20), (1, 3, 20, 20))
- _verify_average((3, 20, 20), (1, 3, 20, 20))
- _verify_average((20, 20), (1, 3, 20, 20))
+ # disable tests for now because ValueError: setting an array element with
a sequence.
+ # The requested array has an inhomogeneous shape after 1 dimensions. The
detected shape
+ # was (2,) + inhomogeneous part.
+ # _verify_average((3, 20, 20), (1, 3, 20, 20))
+ # _verify_average((20, 20), (1, 3, 20, 20))
def _verify_max(input_dim):
diff --git a/tests/python/relax/test_frontend_stablehlo.py
b/tests/python/relax/test_frontend_stablehlo.py
index 667953ab73..63defaf7d7 100644
--- a/tests/python/relax/test_frontend_stablehlo.py
+++ b/tests/python/relax/test_frontend_stablehlo.py
@@ -196,10 +196,6 @@ def test_add_dynamic():
@tvm.testing.requires_gpu
[email protected](
- reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN
library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
def test_unary():
import jax
@@ -233,10 +229,6 @@ def test_unary():
@tvm.testing.requires_gpu
[email protected](
- reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN
library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
def test_binary():
import jax
@@ -258,10 +250,6 @@ def test_binary():
@tvm.testing.requires_gpu
[email protected](
- reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN
library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
def test_const():
import jax
@@ -272,10 +260,6 @@ def test_const():
@tvm.testing.requires_gpu
[email protected](
- reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN
library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
def test_maximum():
import jax
import jax.numpy as jnp
@@ -287,10 +271,6 @@ def test_maximum():
@tvm.testing.requires_gpu
[email protected](
- reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN
library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
def test_minimum():
import jax
import jax.numpy as jnp
@@ -332,10 +312,6 @@ def test_reduce_window():
@tvm.testing.requires_gpu
[email protected](
- reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN
library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
def test_dot_general():
import jax
diff --git a/tests/scripts/task_python_docs.sh
b/tests/scripts/task_python_docs.sh
index 2a213ddd18..dca9c364e4 100755
--- a/tests/scripts/task_python_docs.sh
+++ b/tests/scripts/task_python_docs.sh
@@ -86,6 +86,8 @@ IGNORED_WARNINGS=(
'autotvm:Cannot find config for target=llvm -keys=cpu'
'autotvm:One or more operators have not been tuned. Please tune your model
for better performance. Use DEBUG logging level to see more details.'
'autotvm:Cannot find config for target=cuda -keys=cuda,gpu'
+ 'cannot cache unpickable configuration value:'
+ 'Invalid configuration value found: 'language = None'.'
# Warning is thrown during TFLite quantization for micro_train tutorial
'absl:For model inputs containing unsupported operations which cannot be
quantized, the `inference_input_type` attribute will default to the original
type.'
'absl:Found untraced functions such as _jit_compiled_convolution_op'
diff --git a/tests/scripts/task_python_frontend.sh
b/tests/scripts/task_python_frontend.sh
index ee6be87b36..593e8f50c1 100755
--- a/tests/scripts/task_python_frontend.sh
+++ b/tests/scripts/task_python_frontend.sh
@@ -31,9 +31,6 @@ find . -type f -path "*.pyc" | xargs rm -f
make cython3
-echo "Running relay MXNet frontend test..."
-run_pytest cython python-frontend-mxnet tests/python/frontend/mxnet
-
echo "Running relay ONNX frontend test..."
run_pytest cython python-frontend-onnx tests/python/frontend/onnx
@@ -58,6 +55,3 @@ run_pytest cython python-frontend-paddlepaddle
tests/python/frontend/paddlepaddl
echo "Running relay CoreML frontend test..."
run_pytest cython python-frontend-coreml tests/python/frontend/coreml
-
-echo "Running relay OneFlow frontend test..."
-run_pytest cython python-frontend-oneflow tests/python/frontend/oneflow
diff --git a/tests/scripts/task_python_frontend_cpu.sh
b/tests/scripts/task_python_frontend_cpu.sh
index 52c3d1078e..aac554bea5 100755
--- a/tests/scripts/task_python_frontend_cpu.sh
+++ b/tests/scripts/task_python_frontend_cpu.sh
@@ -36,6 +36,3 @@ run_pytest cython python-frontend-tflite
tests/python/frontend/tflite
echo "Running relay Keras frontend test..."
run_pytest cython python-frontend-keras tests/python/frontend/keras
-
-echo "Running relay Caffe frontend test..."
-run_pytest cython python-frontend-caffe tests/python/frontend/caffe
![https://github.com/dmlc/mxnet.js/blob/main/data/cat.png?raw=true"](https://github.com/dmlc/mxnet.js/blob/main/data/cat.png?raw=true"){kind=link}