triton-ascend-case-elemwise-cast

triton-ascend-case-elemwise-cast is a skill for Claude Code, Codex from mindspore-ai/akg. It costs 68 tokens per session (675 once invoked), scanned A, original, Apache-2.0.

An optimization pattern for converting large arrays from int8 numbers to fp16 numbers on Ascend hardware. It splits the work into blocks and smaller tiles so processing can use the available on-chip memory.

In plain words
What is it for?
Use it for large element-by-element type conversions, especially int8-to-fp16 conversions involving millions of elements.
Why use it?
Large arrays may not fit in on-chip memory at once, while too few blocks can reduce parallelism. The pattern helps balance tile size, memory use, and the number of blocks.

Skill for Claude CodeCodex

Written for no agent in particular: nothing here depends on one.

Good fit Use it for large element-by-element type conversions, especially int8-to-fp16 conversions involving millions of elements.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/mindspore-ai/akg/triton-ascend-case-elemwise-cast
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add mindspore-ai/akg --skill triton-ascend-case-elemwise-cast
Clone the repo
git clone --depth 1 https://github.com/mindspore-ai/akg

Made for: Claude Code, Codex.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for triton-ascend-case-elemwise-cast

README.md
[![agentmods](https://agentmods.dev/badge/skills/mindspore-ai/akg/triton-ascend-case-elemwise-cast/github.svg)](https://agentmods.dev/skills/mindspore-ai/akg/triton-ascend-case-elemwise-cast)
Your own site
<a href="https://agentmods.dev/skills/mindspore-ai/akg/triton-ascend-case-elemwise-cast"><img src="https://agentmods.dev/badge/skills/mindspore-ai/akg/triton-ascend-case-elemwise-cast/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for triton-ascend-case-elemwise-cast

Your own site · 80×15
<a href="https://agentmods.dev/skills/mindspore-ai/akg/triton-ascend-case-elemwise-cast"><img src="https://agentmods.dev/badge/skills/mindspore-ai/akg/triton-ascend-case-elemwise-cast.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 68 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 675 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe. Third-party audits
  • NVIDIA SkillSpector pass 7 Sept 2026
How audits are shown
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00068 $0.00675
Opus 5 $0.00034 $0.00338
Sonnet 5 $0.00014 $0.00135
Haiku 4.5 $0.00007 $0.00068

Measured 11d ago against content hash 48b16dc72d35, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-11, from the pricing page.

Security

Grade A, and why

triton-ascend-case-elemwise-cast scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 11d ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

akg_agents/python/akg_agents/op/resources/skills/triton-ascend/cases/triton-ascend-case-elemwise-cast/SKILL.md · 50 lines

What it actually says

Int8 到 FP16 类型转换优化案例

任务特征

  • 操作类型:Elementwise,类型转换操作
  • 数据尺寸:(128, 1024, 1024),shape较大
  • 数据类型:输入int8,输出fp16
  • 任务特点:可以按照轴的顺序(可flatten为一根轴),外层并行,内层向量化,若UB存不下,可考虑多次切分

优化:二次切分 + 用满UB

# Triton 内核实现:将BLOCK_SIZE分块,每次搬运TILE_SIZE大小的数据
configs = [
    triton.Config({"BLOCK_SIZE": 65536, "TILE_SIZE": 65536}), # 核数2048, 性能最优!用满UB且无二次切分
    triton.Config({"BLOCK_SIZE": 65536, "TILE_SIZE": 32768}), # 核数2048,但UB未用满
    triton.Config({"BLOCK_SIZE": 2097152, "TILE_SIZE": 65536}), # 核数64, 并行度低
    triton.Config({"BLOCK_SIZE": 4194304, "TILE_SIZE": 65536}), # 核数32, 并行度更低
]

# 内核操作:
block_start = pid * BLOCK_SIZE
for i in range(0, BLOCK_SIZE, TILE_SIZE):
    offsets = block_start + tl.arange(0, TILE_SIZE)
    mask = offsets < n_elements
    input_data = tl.load(input_ptr + offsets, mask=mask)
    output_data = tl.cast(input_data, tl.float16)
    tl.store(output_ptr + offsets, output_data, mask=mask)

优化内容

  • triton 内核部分使用for循环,尝试进行二次切分,每次搬运TILE_SIZE大小的数据,提高UB的利用率
  • 在一定范围内提高核数,并尝试用满UB
  • 核内没有二次切分时性能最优(BLOCK_SIZE = TILE_SIZE = 65536)

总结

  1. 当数据的shape较大时,为了获得更佳的性能,切分值设置尽量能被shape的大小整除
  2. 对于单纯的Elementwise操作,将多根轴的元素展开为一根轴,然后在这根轴上进行切分
  3. 将block分配给每个线程块,若UB存不下,可考虑多次切分(二次切分)
Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 11d ago First seen · 50 lines · 68 tokens per session scan A 48b16dc72d35

Subscribe to this mod's changes

triton-ascend-case-elemwise-cast is a skill published in the GitHub repository mindspore-ai/akg (259 stars, last pushed 1mo ago), licensed Apache-2.0. It adds 68 tokens to every session and 675 once invoked, about $0.0003 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.

Related

Other skills, from other repositories

triton-ascend-case-matmul-swizzle2d

A case study for optimizing matrix multiplication in Triton on Ascend hardware using 2D block swizzling and a fixed number of cores.

wenyi-li/awesome-agent-kernel-skills · 88 tokens

triton-ascend-case-reduction-weighted-swiglu

A performance-optimization method for a specialized 3D tensor operation using Weighted SwiGLU during backward computation. It reshapes data and divides work to improve parallel processing while keeping within on-chip memory limits.

wenyi-li/awesome-agent-kernel-skills · 97 tokens

triton-ascend-case-elemwise-cast

A performance-tuning skill for converting large arrays from one data type to another, such as int8 to fp16. It uses two levels of splitting to improve use of shared on-chip memory on Ascend hardware.

wenyi-li/awesome-agent-kernel-skills · 68 tokens

triton-ascend-case-elemwise-zeros

An optimization for creating small tensors with zeros, arange, or full in Triton Ascend. Triton Ascend is a programming environment for running tensor operations on Ascend hardware.

wenyi-li/awesome-agent-kernel-skills · 67 tokens

gke-compute-classes

Configures, optimizes, and troubleshoots GKE ComputeClasses. Use when configuring Spot VMs with on-demand fallback, targeting specific accelerators (GPUs/TPUs) or machine families, restricting ComputeClass access, or debugging pending pods related to node pool auto-creation. Do not use for cluster-level Node Auto…

google/skills · 83 tokens

jetson-diagnostic

Read-only Jetson health snapshot for identity, memory, GPU, thermal, power, storage, services, and top processes.

NVIDIA/skills · 30 tokens