search

FlagOpen / FlagGems

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

[Operator] Add repeat op #195

Closed zfu82 closed 2 months ago

zfu82 commented 3 months 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 3 months ago

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

done