Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
gemm_sp.h 203 Bytes
Newer Older
botbw's avatar
[Experimental][Language] add `T.GEMM_SP` for sm90 sparse tensor core (#526)  
botbw committed
1
2
3 
#pragma once
#if (defined(__CUDA_ARCH_LIST__) && (__CUDA_ARCH_LIST__ >= 900))
#include "gemm_sp_sm90.h"
botbw's avatar
[feat] support gemm_sp for ampere and ada arch (#691)  
botbw committed
4
5 
#else(defined(__CUDA_ARCH_LIST__) && (__CUDA_ARCH_LIST__ >= 800))
#include "gemm_sp_sm80.h"
botbw's avatar
[Experimental][Language] add `T.GEMM_SP` for sm90 sparse tensor core (#526)  
botbw committed
6 
#endif