Fix type consistency in SingleTileWithTMALoadStoreStMatrix for fusion.manage.

[csarofeen]

This was causing errors in one of my development branches. We need to be consistent with types that we manage with fusion when storing and getting.

[csarofeen]

Same issue as seen in #3520

[naoyam]

@protonu Can you please change the type of tile_sizes to int64_t.

https://github.com/NVIDIA/Fuser/blob/main/tests/cpp/test_mma.cpp#L418

We just use int64_t for any integer data unless required to be other integer types. The cast shouldn't be necessary with this change.

[protonu]

I'll accept this change and send a new PR to fix the type.

[protonu]

!test

[zasdfgbnm]

Just realized that my PR hasn’t been merged yet: #3521
I believe this PR will ensure consistency of type during cloning, and push the error to the point when we get the managed data in lowering.

[protonu]

Just realized that my PR hasn’t been merged yet: #3521 I believe this PR will ensure consistency of type during cloning, and push the error to the point when we get the managed data in lowering.

@zasdfgbnm I missed your PR. I landed this a minute ago (#3614). You may need to rebase.