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 repeat op #195

Closed zfu82 closed 1 month ago

zfu82 commented 1 month ago

PR Category

Operator

Type of Change

New Feature

Description

Add repeat op.

Issue

similar to https://github.com/FlagOpen/FlagGems/pull/148

Progress

Performance

Tested on NV-A100 image

zfu82 commented 1 month ago

Where to put the codegen file and why the the test does not coverage the repeat.py file

done