[PR] [BugTIR]fix error merging shared memory for ptx_cp_async [tvm]

Wed, 27 Mar 2024 00:13:18 -0700 


JackWeiw opened a new pull request, #16800:
URL: https://github.com/apache/tvm/pull/16800

   In driver_api.cc,  
   ```
   // MergeSharedMemoryAllocations must be applied after SplitHostDevice
    // because the merged allocation site is at the beginning of each device 
function
   ```
   howerver MergeSharedMemoryAllocations didn't take care of call op 
`ptx_cp_async`,
   result in error
   [script to 
reproduce](https://gist.github.com/JackWeiw/d7c804744dc7fa0fc17d6d6f69997057) 
   
![image](https://github.com/apache/tvm/assets/126441921/14fbb7a5-9a88-4ae6-b5c1-55a636824e51)
   as we can see it didn't successfully merge shared memory in `ptx_cp_async`
   
![image](https://github.com/apache/tvm/assets/126441921/76c3aef2-6da8-4bc1-a442-e1740a514429)
   


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to