Commit Graph

3 Commits

Author SHA1 Message Date
yangzhenzhang 4ef8e25944 modify vmap rule for masked select 2022-05-26 16:11:32 +08:00
yangzhenzhang 3b187616e7 add vmap for masked select 2022-05-23 09:10:35 +08:00
yangzhenzhang c7a59aa805 add gpu impl of masked select 2022-05-18 09:14:07 +08:00