(tvm) branch test_all_cases_on_unity updated: upd

[syfeng]

This is an automated email from the ASF dual-hosted git repository.

syfeng pushed a commit to branch test_all_cases_on_unity
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/test_all_cases_on_unity by 
this push:
     new bd9a21f9c9 upd
bd9a21f9c9 is described below

commit bd9a21f9c9656fa518d69df07f327ba95aba1ee2
Author: Siyuan Feng <hzfen...@sjtu.edu.cn>
AuthorDate: Mon Dec 4 16:26:59 2023 +0800

    upd
---
 ci/jenkins/generated/cpu_jenkinsfile.groovy |  3 ++-
 include/tvm/topi/transform.h                | 11 +++++++++--
 rust/tvm/src/ir/module.rs                   |  6 +++---
 3 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/ci/jenkins/generated/cpu_jenkinsfile.groovy 
b/ci/jenkins/generated/cpu_jenkinsfile.groovy
index 38364747bc..3a5f888427 100644
--- a/ci/jenkins/generated/cpu_jenkinsfile.groovy
+++ b/ci/jenkins/generated/cpu_jenkinsfile.groovy
@@ -572,7 +572,8 @@ def build(node_type) {
         ci_setup(ci_cpu)
         // sh "${docker_run} ${ci_cpu} ./tests/scripts/task_golang.sh"
         // TODO(@jroesch): need to resolve CI issue will turn back on in 
follow up patch
-        sh (script: "${docker_run} ${ci_cpu} ./tests/scripts/task_rust.sh", 
label: 'Rust build and test')
+
+        // sh (script: "${docker_run} ${ci_cpu} ./tests/scripts/task_rust.sh", 
label: 'Rust build and test')
             })
           }
         }
diff --git a/include/tvm/topi/transform.h b/include/tvm/topi/transform.h
index 009f8cdd30..7ee09d834f 100644
--- a/include/tvm/topi/transform.h
+++ b/include/tvm/topi/transform.h
@@ -43,6 +43,8 @@
 #include <unordered_set>
 #include <vector>
 
+#include "tvm/tir/expr.h"
+
 namespace tvm {
 namespace topi {
 
@@ -716,8 +718,13 @@ inline Tensor dynamic_strided_slice(const Tensor& x, const 
Array<PrimExpr>& begi
 
   arith::Analyzer analyzer;
   for (size_t i = 0; i < num_slice_axes; ++i) {
-    auto d = analyzer.Simplify(indexdiv(end[i] - begin[i], strides[i]));
-    out_shape.push_back(d);
+    // Check ProducerLoad to keep backward compatibility for Relay.
+    if (!begin[i]->IsInstance<ProducerLoadNode>() && 
!end[i]->IsInstance<ProducerLoadNode>() &&
+        !strides[i]->IsInstance<ProducerLoadNode>()) {
+      out_shape.push_back(analyzer.Simplify(indexdiv(end[i] - begin[i], 
strides[i])));
+    } else {
+      out_shape.push_back(tvm::tir::Var("dim"));
+    }
   }
 
   for (size_t i = num_slice_axes; i < src_tensor_dim; ++i) {
diff --git a/rust/tvm/src/ir/module.rs b/rust/tvm/src/ir/module.rs
index a4eccbbf41..b78c9e4a9c 100644
--- a/rust/tvm/src/ir/module.rs
+++ b/rust/tvm/src/ir/module.rs
@@ -62,7 +62,7 @@ external! {
     #[name("relay.parser.ParseExpr")]
     fn parse_expression(file_name: TVMString, source: TVMString) -> IRModule;
     #[name("ir.IRModule")]
-    fn module_new(funcs: Map<GlobalVar, BaseFunc>, types: Map<GlobalTypeVar, 
TypeData>, attrs: Map<TVMString, ObjectRef>, global_infos: Map<TVMString, 
Array<ObjectRef>>) -> IRModule;
+    fn module_new(funcs: Map<GlobalVar, BaseFunc>, types: Map<GlobalTypeVar, 
TypeData>, attrs: DictAttrs, global_infos: Map<TVMString, Array<ObjectRef>>) -> 
IRModule;
     // Module methods
     #[name("ir.Module_Add")]
     fn module_add(module: IRModule, type_name: GlobalVar, expr: BaseFunc, 
update: bool) -> IRModule;
@@ -109,7 +109,7 @@ impl IRModule {
         module_new(
             Map::from_iter(funcs),
             Map::from_iter(types),
-            Map::from_iter(attrs),
+            attrs,
             Map::from_iter(global_infos),
         )
     }
@@ -123,7 +123,7 @@ impl IRModule {
             funcs.iter(),
             types.iter(),
             attrs.iter(),
-            global_infos.iter()
+            global_infos.iter(),
         )
     }