File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -840,7 +840,7 @@ def env_thread(env_name, span):
840840 self .context .report_error (
841841 f"VarDef expected assign to only one var, but got { names } " , span
842842 )
843- v = Var (names [0 ], span = span )
843+ v = Var (names [0 ], dtype = "int32" , span = span )
844844 self .context .func_var_env_dict [v ] = env_name
845845 self .context .update_symbol (v .name , v , self .node )
846846
Original file line number Diff line number Diff line change 1515# specific language governing permissions and limitations
1616# under the License.
1717# pylint: disable=invalid-name
18- """Intrinsics for x86 tensorization."""
19- from .. import TensorIntrin
18+ """Intrinsics for ARM tensorization."""
2019from tvm .script import tir as T
20+ from .. import TensorIntrin
2121
2222
2323# TODO(masahi): Parametrize the TVMScript description of dot product by
You can’t perform that action at this time.
0 commit comments