Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
gemm.h 385 Bytes
Newer Older
Lei Wang's avatar
[Initialization] Migration of Codebase from Dev Branch into Main (#10)  
Lei Wang committed
1
2
3 
#pragma once
#if (defined(__CUDA_ARCH_LIST__) && (__CUDA_ARCH_LIST__ >= 900))
#include "gemm_sm90.h"
Lei Wang's avatar
[Enhancement] Add support for CUDA architecture 8.9 in GEMM template (#304)  
Lei Wang committed
4
5 
#elif (defined(__CUDA_ARCH_LIST__) && (__CUDA_ARCH_LIST__ >= 890))
#include "gemm_sm89.h"
Lei Wang's avatar
[Initialization] Migration of Codebase from Dev Branch into Main (#10)  
Lei Wang committed
6
7
8
9
10
11
12 
#elif (defined(__CUDA_ARCH_LIST__) && (__CUDA_ARCH_LIST__ >= 750))
#include "gemm_sm80.h"
#elif (defined(__CUDA_ARCH_LIST__) && (__CUDA_ARCH_LIST__ >= 700))
#include "gemm_sm70.h"
#else

#endif