[MLIR][NVVM] Fix wmma test after d3edc94d (#170659)

See discussion on #169061
This commit is contained in:
Sohaib Iftikhar 2025-12-04 14:37:29 +01:00 committed by GitHub
parent 04bc0af103
commit b31a398bcf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -44,6 +44,7 @@ func.func @main() {
gpu.host_register %2 : memref<*xf64>
gpu.host_register %20 : memref<*xf64>
gpu.host_register %33 : memref<*xf64>
gpu.host_register %34 : memref<*xf64>
gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %c1, %grid_y = %c1, %grid_z = %c1)
threads(%tx, %ty, %tz) in (%block_x = %c32, %block_y = %c1, %block_z = %c1) {