mindspore-ai

60 mods across 1 repository, 259 stars between them.

hint-mode

25

mindspore-ai/akg

Skill Claude CodeCodex

A guide for describing tunable parameters in a standard hint format. It turns value lists, numeric ranges, fixed values, and powers of two into configuration entries for later automatic tuning.

259 23d ago A 33 tokens original Apache-2.0

sketch-design

26

mindspore-ai/akg

Skill Claude CodeCodex

A small language and method for sketching how a computing operation should work before writing its implementation. It describes tensors, memory actions, loops, and optimization hints in a compact form.

259 23d ago A 30 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A workflow guide for KernelAgent, an assistant that creates and checks computing operations through a step-by-step ReAct process. It requires the user to confirm the interpretation, generated task, execution method, and result.

259 23d ago A 14 tokens original Apache-2.0

kernel-workflow

28

mindspore-ai/akg

Skill Claude CodeCodex

A workflow for generating, checking, and optimizing GPU kernels from a task description or existing code. A kernel is a GPU function designed to process data in parallel.

259 23d ago A 50 tokens original Apache-2.0

performance-summary

29

mindspore-ai/akg

Skill Claude CodeCodex

A skill for combining performance results from several operators, which are reusable computation steps such as ReLU or sigmoid. It reads recorded measurements and produces a comparison of latency, throughput, memory use, and speedup.

259 23d ago A 48 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A PyPTO example of GELU, an activation function that smoothly scales neural-network values instead of simply keeping or discarding them. It applies the calculation element by element after flattening the input.

259 23d ago A 40 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

An example of a PyPTO kernel for calculating cross-entropy loss, a common machine-learning measure of prediction error. It processes predictions and target classes in tiles, applies softmax, selects the target probabilities, and returns one scalar loss.

259 23d ago A 37 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

An example of a two-dimensional matrix multiplication kernel using loop-based blocks and separate handling for leftover rows. Matrix multiplication combines rows and columns of number grids to produce a new grid.

259 23d ago A 34 tokens original Apache-2.0

pypto-case-matvec

33

mindspore-ai/akg

Skill Claude CodeCodex

A PyPTO pattern for multiplying a matrix by a column vector when the vector length is over 65,535. It reshapes the vector into a row and sums element-by-element products.

259 23d ago A 32 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A PyPTO example of BatchNorm, a method that normalizes values using statistics calculated across a batch and spatial positions. It reshapes the input to three dimensions and processes channels in groups.

259 23d ago A 41 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A PyPTO example of LayerNorm, which normalizes each row of values using that row’s mean and variance. It flattens each input item to two dimensions, processes rows in groups, and restores the original shape.

259 23d ago A 42 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A PyPTO example that adds values along one dimension of a three-dimensional tensor while keeping that dimension in the output. Reduction means combining many values into one, such as with a sum.

259 23d ago A 30 tokens original Apache-2.0

pypto-api

37

mindspore-ai/akg

Skill Claude CodeCodex

A quick reference for PyPTO's programming interface, including kernel decorators, tensors, data types, and tile configuration. PyPTO is a Python interface for describing tensor and kernel computations.

259 23d ago A 18 tokens original Apache-2.0

pypto-basics

38

mindspore-ai/akg

Skill Claude CodeCodex

A set of rules for implementing PyPTO kernels from PyTorch code. It starts by writing down the original operation's exact mathematical meaning before choosing an implementation or optimization.

259 23d ago A 14 tokens original Apache-2.0

pypto-loop-view

39

mindspore-ai/akg

Skill Claude CodeCodex

A rulebook for using PyPTO loops and views when view shapes must be known during compilation. A view is a different way of interpreting part of a tensor without changing its data.

259 23d ago A 43 tokens original Apache-2.0

pypto-optimization

40

mindspore-ai/akg

Skill Claude CodeCodex

A set of PyPTO performance-tuning rules for tile sizes, loops, data reduction, and operations such as softmax, normalization, and loss calculations. A tile is a small block of tensor data processed together.

259 23d ago A 63 tokens original Apache-2.0

pypto-pitfalls

41

mindspore-ai/akg

Skill Claude CodeCodex

A list of PyPTO coding pitfalls and corrected patterns for tensor arithmetic, clamping, loss functions, factory parameters, and large matrix multiplications. PyPTO is a tensor-programming system with rules that differ from ordinary Python arithmetic.

259 23d ago A 18 tokens original Apache-2.0

task-constructor

42

mindspore-ai/akg

Skill Claude CodeCodex

A workflow for extracting a PyTorch or Triton operation from a code repository and packaging it as one self-contained KernelBench task file. KernelBench is a format for standardizing and testing machine-learning kernel tasks.

259 23d ago A 81 tokens original Apache-2.0

tilelang-cuda-api

43

mindspore-ai/akg

Skill Claude CodeCodex

A reference manual for the TileLang CUDA programming interface, including kernel compilation, launch dimensions, thread blocks, memory allocation, and utility functions. TileLang is a Python-like language for writing GPU kernels.

259 23d ago A 43 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A beginner-to-advanced guide to TileLang CUDA, covering its Python-like syntax, GPU kernels, thread blocks, and memory types. A kernel is a function compiled to run many operations in parallel on a GPU.

259 23d ago A 24 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A collection of runnable examples showing how to write CUDA kernels in TileLang, a Python-like language for GPU code, and use them with PyTorch, a machine-learning library.

259 23d ago A 20 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A guide to improving how TileLang CUDA kernels move data through GPU memory. It covers shared memory, registers, local storage, data layouts, combined memory access, and avoiding bank conflicts, where threads compete for the same memory bank.

259 23d ago A 62 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A reference guide for optimizing and debugging TileLang programs that run CUDA GPU kernels. TileLang is a programming language for describing GPU computations.

259 23d ago A 64 tokens original Apache-2.0

mindspore-ai/akg

Skill Claude CodeCodex

A collection of standard TileLang code patterns and templates for common GPU computations. It explains how to write element-wise operations, reductions, matrix multiplication, and matrix–vector multiplication.

259 23d ago A 72 tokens original Apache-2.0

At most 3 mods per repository are shown here, and a mod shipped inside a plugin is left to that plugin's page — the rest are on their repository pages: