search

FlagOpen / FlagGems

FlagGems is an operator library for large language models implemented in Triton Language.
Apache License 2.0
296 stars 27 forks source link

[Operator] Add isin op #188

Closed zhzhcookie closed 1 month ago

zhzhcookie commented 1 month ago
  1. Add isin op.
  2. Add unit tests and performance tests of isin op.
  3. Optimize the implement of unique op, which is used by isin op.