triton-ascend-case-elemwise-broadcast-2d

Community

Optimize 2D broadcast ops on Ascend with Triton.

Authorxchang1121
Version1.0.0
Installs0

System Documentation

What problem does it solve?

2D broadcast division optimization for Ascend-backed Triton kernels, reducing fragmentation and improving UB utilization on large broadcast axes while keeping non-broadcast axes efficient.

Core Features & Use Cases

  • 2D broadcast optimization with full-dimension handling when NUM_BLOCKS is fixed
  • Efficient data reuse by loading divisors once and broadcasting within the kernel
  • Use Case: accelerate large-scale 2D broadcast computations on Ascend devices using Triton

Quick Start

Run the Triton Ascend-optimized 2D broadcast kernel with appropriate NUM_BLOCKS and SUB_M settings to benchmark performance.

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-broadcast-2d
Download link: https://github.com/xchang1121/AutoResearch-CC-hook/archive/main.zip#triton-ascend-case-elemwise-broadcast-2d

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.