Skip to content

Gao-HaoYuan/TensorOps

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

This is a component specifically designed for tensor kernel. It draws inspiration from Torch and Cutlass, processing higher-level Tensors into Layout class, and invoking device resources through the functin gpu_kernel. It also mimics torch’s Half, but unlike torch, it directly supports computation with CUDA’s __half type.

About

A component specifically designed for tensor kernel.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors