[Bug][MetaSchedule] Compute-At DataType Mismatch

[zxybazh]

I noticed a data type issue that can be triggered when applying compute-at, causing invalid expression in TIR script with incompatible datatype, thus failure in post processors and build.

To reproduce simple run this script would work. It can build successfully before schedule but fails with a data type mismatch error after a compute-at scheduling.

cc @Hzfengsy @junrushao1994

[masahi]

Does this issue still exist? Your test script doesn't work due to the MS API change.

[zxybazh]

Yes, the issue is still there, I changed the script (remove TuneConfig import) and it can work now.

[masahi]

The PR #13327 works at CreatePrimFunc level and would change the generated TVMScript module. So we need to regenerate this TVMScript to verify that the PR fixes this issue.

[zxybazh]

Cannot reproduce, close for now.