Name		Name	Last commit message	Last commit date
benchmark		benchmark
include/runtime		include/runtime
kernels		kernels
src		src
.gitignore		.gitignore
CMakeLists.txt		CMakeLists.txt
README.md		README.md

Repository files navigation

小型深度学习算子的Kernel实现库

1.项目简介

本项目将会实现常见的深度学习模型中的算子。
算子大致包括:卷积，池化，激活，Softmax，Attention，LayerNorm等
实现方式包括:cuda kernel，cublas/cudnn库，cutlass库等

About

No description, website, or topics provided.

0 stars

0 watching

Report repository

Releases

Packages

Contributors

Languages