Move graph definitions to python string, Change Relax tests to struct…

…ural, Move Relay tests to nightly
apache · Jun 7, 2024 · 4b3c838 · 4b3c838
1 parent e70842e
commit 4b3c838
Show file tree

Hide file tree

Showing 307 changed files with 5,612 additions and 3,735 deletions.
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
@@ -17,7 +17,7 @@
 
 # CI docker GPU env
 # tag: v0.60
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
 
 COPY utils/apt-install-and-clear.sh /usr/local/bin/apt-install-and-clear
 

diff --git a/python/tvm/relax/frontend/nnef/nnef_ops.py b/python/tvm/relax/frontend/nnef/nnef_ops.py
@@ -1419,7 +1419,7 @@ def pad_converter(bbuilder, data, padding, border, value, **kwargs):
         return bbuilder.emit_te(tvm.topi.nn.mirror_pad, data, pad_before, pad_after, "REFLECT")
     if border == "edge":
         raise tvm.error.OpNotImplemented(
-            "Replicate - Edge mode is currently not supperted in TVM relax"
+            "Replicate - Edge mode is currently not supported in TVM relax"
         )
 
     # constant works with normal relax.nn.pad

diff --git a/tests/python/frontend/nnef/cases/abs_2d/graph.nnef b/tests/python/frontend/nnef/cases/abs_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/abs_4d/graph.nnef b/tests/python/frontend/nnef/cases/abs_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/acos_2d/graph.nnef b/tests/python/frontend/nnef/cases/acos_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/acos_4d/graph.nnef b/tests/python/frontend/nnef/cases/acos_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/acosh_2d/graph.nnef b/tests/python/frontend/nnef/cases/acosh_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/acosh_4d/graph.nnef b/tests/python/frontend/nnef/cases/acosh_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/add_2d/graph.nnef b/tests/python/frontend/nnef/cases/add_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/add_4d/graph.nnef b/tests/python/frontend/nnef/cases/add_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/add_4d_broadcast/graph.nnef b/tests/python/frontend/nnef/cases/add_4d_broadcast/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/add_4d_constant/graph.nnef b/tests/python/frontend/nnef/cases/add_4d_constant/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/all_reduce_channel/graph.nnef b/tests/python/frontend/nnef/cases/all_reduce_channel/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/all_reduce_spatial/graph.nnef b/tests/python/frontend/nnef/cases/all_reduce_spatial/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/and_2d/graph.nnef b/tests/python/frontend/nnef/cases/and_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/and_4d/graph.nnef b/tests/python/frontend/nnef/cases/and_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/and_4d_broadcast/graph.nnef b/tests/python/frontend/nnef/cases/and_4d_broadcast/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/and_4d_constant/graph.nnef b/tests/python/frontend/nnef/cases/and_4d_constant/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/any_reduce_channel/graph.nnef b/tests/python/frontend/nnef/cases/any_reduce_channel/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/any_reduce_spatial/graph.nnef b/tests/python/frontend/nnef/cases/any_reduce_spatial/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/area_downsample/graph.nnef b/tests/python/frontend/nnef/cases/area_downsample/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/argmax_reduce_channel/graph.nnef b/tests/python/frontend/nnef/cases/argmax_reduce_channel/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/argmax_reduce_spatial/graph.nnef b/tests/python/frontend/nnef/cases/argmax_reduce_spatial/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/argmin_reduce_channel/graph.nnef b/tests/python/frontend/nnef/cases/argmin_reduce_channel/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/argmin_reduce_spatial/graph.nnef b/tests/python/frontend/nnef/cases/argmin_reduce_spatial/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/asin_2d/graph.nnef b/tests/python/frontend/nnef/cases/asin_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/asin_4d/graph.nnef b/tests/python/frontend/nnef/cases/asin_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/asinh_2d/graph.nnef b/tests/python/frontend/nnef/cases/asinh_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/asinh_4d/graph.nnef b/tests/python/frontend/nnef/cases/asinh_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/atan_2d/graph.nnef b/tests/python/frontend/nnef/cases/atan_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/atan_4d/graph.nnef b/tests/python/frontend/nnef/cases/atan_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/atanh_2d/graph.nnef b/tests/python/frontend/nnef/cases/atanh_2d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/atanh_4d/graph.nnef b/tests/python/frontend/nnef/cases/atanh_4d/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool1x1/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool1x1/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool2x2/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool2x2/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool3x3/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool3x3/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool3x3_ignore-border/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool3x3_ignore-border/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool3x3_pad0-0/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool3x3_pad0-0/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool3x3_pad0-1/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool3x3_pad0-1/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool3x3_pad1-0/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool3x3_pad1-0/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool3x3_pad1-1/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool3x3_pad1-1/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/avg_pool3x3_stride1x1/graph.nnef b/tests/python/frontend/nnef/cases/avg_pool3x3_stride1x1/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/batch_norm/graph.nnef b/tests/python/frontend/nnef/cases/batch_norm/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/bilinear_upsample_aligned_constant/graph.nnef b/tests/python/frontend/nnef/cases/bilinear_upsample_aligned_constant/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/bilinear_upsample_aligned_replicate/graph.nnef b/tests/python/frontend/nnef/cases/bilinear_upsample_aligned_replicate/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/bilinear_upsample_asymmetric_constant/graph.nnef b/tests/python/frontend/nnef/cases/bilinear_upsample_asymmetric_constant/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/bilinear_upsample_asymmetric_replicate/graph.nnef b/tests/python/frontend/nnef/cases/bilinear_upsample_asymmetric_replicate/graph.nnef
diff --git a/tests/python/frontend/nnef/cases/bilinear_upsample_symmetric_constant/graph.nnef b/tests/python/frontend/nnef/cases/bilinear_upsample_symmetric_constant/graph.nnef