https://github.com/DeMoriarty/custom_matmul_kernels

Customized matrix multiplication kernels
https://github.com/DeMoriarty/custom_matmul_kernels

Last synced: 11 days ago
JSON representation

Customized matrix multiplication kernels

• Host: GitHub
• URL: https://github.com/DeMoriarty/custom_matmul_kernels
• Owner: DeMoriarty
• License: gpl-3.0
• Created: 2021-05-02T09:38:07.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2022-03-05T10:42:03.000Z (over 2 years ago)
• Last Synced: 2024-10-28T09:20:22.894Z (13 days ago)
• Language: Jupyter Notebook
• Homepage:
• Size: 2.55 MB
• Stars: 53
• Watchers: 1
• Forks: 6
• Open Issues: 4
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Custom Matmul Kernels

This repository contains source code for [this blog post](https://demoriarty.github.io/BMM-1/).

## Dependency

- Python 3.7.10 or higher

- CuPy 7.4.0 or higher

- Pytorch 1.8.1 or higher

- Only tested with CUDA 11.2