triton-ascend-case-elemwise-zeros

Community

Small-shape zeros optimization on Ascend

Authorxchang1121
Version1.0.0
Installs0

System Documentation

What problem does it solve?

小shape张量创建(zeros/arange/full)优化:通过减少核数避免多核启动和调度开销,单核处理性能优于多核并行,适用于shape较小(千级元素)的elementwise创建张量场景

Core Features & Use Cases

  • 操作类型:elemwise 创建张量(zeros、arange、full 等)
  • 数据尺寸:小型张量,如 2×256×16,float32
  • 优化点:通过调整 BLOCK_SIZE,降低核启动成本,实现单核优于多核并行
  • 使用场景:Ascend 平台的轻量级张量创建路径,尤其在性能敏感的场景中

Quick Start

Run the provided zeros kernel example with a small input to observe reduced kernel launches.

Dependency Matrix

Required Modules

None required

Components

Standard package

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: triton-ascend-case-elemwise-zeros
Download link: https://github.com/xchang1121/AutoResearch-CC-hook/archive/main.zip#triton-ascend-case-elemwise-zeros

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.