Supported OPs - Triton¶
이 페이지는 Triton 실행 환경에서 검증된 supported Triton Ops를 나열합니다.
아래 연산들은 RBLN NPU에서 사용할 수 있습니다. 지원 범위는 시간이 지나면서 확대될 예정입니다.
지원하지 않는 연산을 사용하면 예상치 못한 컴파일 오류가 발생할 수 있습니다.
참고: triton.language (Triton 공식 문서)
Programming Model¶
| Triton OPs | Description |
|---|---|
tensor |
- |
Creation Ops¶
| Triton OPs | Description |
|---|---|
zeros |
- |
Shape Manipulation Ops¶
| Triton OPs | Description |
|---|---|
broadcast |
- |
broadcast_to |
- |
expand_dims |
- |
permute |
- |
reshape |
- |
trans |
- |
Linear Algebra Ops¶
| Triton OPs | Description |
|---|---|
dot |
- |
Memory/Pointer Ops¶
| Triton OPs | Description |
|---|---|
load |
- |
store |
- |
make_block_ptr |
- |
advance |
- |
Indexing Ops¶
| Triton OPs | Description |
|---|---|
where |
- |
Math Ops¶
| Triton OPs | Description |
|---|---|
abs |
- |
ceil |
- |
cos |
- |
erf |
- |
exp |
- |
exp2 |
- |
fdiv |
- |
floor |
- |
log |
- |
log2 |
- |
maximum |
- |
minimum |
- |
rsqrt |
- |
sigmoid |
- |
sin |
- |
sqrt |
- |
Reduction Ops¶
| Triton OPs | Description |
|---|---|
max |
- |
min |
- |
sum |
- |
Iterators¶
| Triton OPs | Description |
|---|---|
static_range |
- |
Debug Ops¶
| Triton OPs | Description |
|---|---|
static_print |
- |
static_assert |
- |