dsl-baseline-generation

Community

Turn PyTorch into AscendDSL baseline code

AuthorJust-it
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This skill enables auto-generation of a baseline AscendDSL implementation from a functional PyTorch operator, streamlining the transition from Python-level models to device-specific DSL code.

Core Features & Use Cases

  • Automatically scaffolds host partitioning and kernel tiling for AscendDSL.
  • Uses example references to guide input/output shapes and implementation patterns.
  • Quick-start path from a PyTorch function to a ready-to-run DSL file.

Quick Start

Place your PyTorch operator file named {op_name}_functional.py and run the generator to produce output/{op_name}/{op_name}_dsl.py.

Dependency Matrix

Required Modules

None required

Components

references

💻 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: dsl-baseline-generation
Download link: https://github.com/Just-it/AscendOpGenAgent/archive/main.zip#dsl-baseline-generation

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.