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

[fix] Fix the input shape of gather's perf test #232

Closed GwokHiujin closed 1 week ago

GwokHiujin commented 1 week ago

As title, fixed the test data shape in the gather performance test that could lead to errors.