pypto-case-matvec

Community

Fast matvec for large K with elementwise tricks

Authorxchang1121
Version1.0.0
Installs0

System Documentation

What problem does it solve?

Solve large-K matrix-vector multiplication by replacing matmul with an elementwise multiply-and-sum strategy.

Core Features & Use Cases

  • Large-K matvec optimization using elementwise multiply and sum.
  • Pypto-based kernel generation for matrix-vector workloads.
  • Use Case: A: (256, 131072), B: (131072, 1) to produce C: (256, 1) with correct results.

Quick Start

Create tensors A with shape (256,131072) and B with shape (131072,1) and execute the matvec kernel to verify the elementwise-sum result for large K.

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: pypto-case-matvec
Download link: https://github.com/xchang1121/AutoResearch-CC-hook/archive/main.zip#pypto-case-matvec

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.