tree b15f645bd5c052389480f9c9f7c0d79aa0d5a896
parent 91433fc05de4500d05a88f4caf76e753ceddf0fc
author Han-Chung Wang <hanhan0912@gmail.com> 1722447829 -0700
committer GitHub <noreply@github.com> 1722447829 -0700
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJmqnfVCRC1aQ7uu5UhlAAAaIYQAIXFocD5MNVuZwQSBKUmTAil
 OgeZZ58SgsFCg9kc1YVr7zyEr0b3XM/yH0O4k7kUvO3bMqC7XHLnO4dvoSyEhvO0
 NkG99sa/n78UTy11I0AtdfmSkRTqsLXKWiDk+XaZOgrw/rS5texYyI7Dakff1RJ1
 CpLKMQ8VPF4dcO2vCxXghg1PsLdMjdtb8jQh4TyxPoK31r5rI+Uh+CHNmJk5AwjT
 PTu3ew5uFtiXu210sIhe37VMIihyBxD3u1zIsKN7Th+N+Zbb12nl4w6TlhmQ4ZOk
 CAM0LM8VS2fOflQ4V45M9SEUldORUzb735vSlxyHRob2pF/V7CPJTppQf4snKdjW
 +xb3t1fxFGHwRHbWU6Mdm4wr28XHZ87gdX2sB+xBYfzwOY8RXk9fqhs/+najGid1
 co8Qit5P3S89uNmoNmloAMBHMDlN3djC1QTUUwTXMzlUDBQOv8dH+CxzofXwEnjI
 mccTwXeuoALZZLjJvYwOIQCyK9LIYeGbkyY3Gs7I7LtbkX7BRkg4Q1VCzRuaypqs
 AQzCnjgZL8WavXybJbWbex1sJNGvWyzzcXtEdlu+fd64ZHEjl8EFfaIyoZaEx2os
 ZjF6HV91+acJ+qdENlIGYrLa9irebdg4UOLL3SCZoAVagFmv/HhS5PKI1y2JYS2I
 DgJLieeaxvIGJRwNag2B
 =RcA3
 -----END PGP SIGNATURE-----
 

[GPU] Fix offsets calculation formula in MultiMmaOp distribution. (#18055)

It was

```
vtid: virtual thread id
tid: lane id
vtid = (tid floordiv stride_i) mod size_i
```

However, it does not take `element` into account. Each thread grabs
`element` contiguous data, so the vtid needs to be multiplied by
`element` to get the next bunch of data. I.e., it becomes

```
vtid: virtual thread id
tid: lane id
vtid = ((tid floordiv stride_i) mod size_i) * element_i
```

Fixes https://github.com/iree-org/iree/issues/17973

---------

Signed-off-by: hanhanW <hanhan0912@gmail.com>