cutlass 3.9 update (#2255)
* cutlass 3.9 update * rebase * fixes out of shared memory for blockwise Blackwell * doc format * fix issue 2253 * disable host ref by default * fix sm120 smem capacity --------- Co-authored-by: yuzhai <yuzhai@nvidia.com> Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
This commit is contained in:
co-authored by
yuzhai
Haicheng Wu
parent
8e345c5c5b
commit
331a1f5b3f
@@ -53,6 +53,20 @@
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#ifndef CUTLASS_GDC_ENABLED
|
||||
#if(CUDA_BARRIER_ENABLED && \
|
||||
defined(CUTLASS_ENABLE_GDC_FOR_SM100) && \
|
||||
defined(__CUDA_ARCH__) && \
|
||||
((__CUDA_ARCH__ == 1000 &&\
|
||||
(defined(__CUDA_ARCH_FEAT_SM100_ALL) || CUDA_ARCH_FAMILY(1000))) || \
|
||||
(__CUDA_ARCH__ == 1010 &&\
|
||||
(defined(__CUDA_ARCH_FEAT_SM101_ALL) || CUDA_ARCH_FAMILY(1010))) || \
|
||||
(__CUDA_ARCH__ == 1200 &&\
|
||||
(defined(__CUDA_ARCH_FEAT_SM120_ALL) || CUDA_ARCH_FAMILY(1200)))))
|
||||
#define CUTLASS_GDC_ENABLED
|
||||
#endif
|
||||
#endif
|
||||
|
||||
namespace cutlass {
|
||||
namespace arch {
|
||||
|
||||
@@ -84,6 +98,5 @@ static constexpr bool IsGdcGloballyEnabled = true;
|
||||
static constexpr bool IsGdcGloballyEnabled = false;
|
||||
#endif
|
||||
|
||||
|
||||
} // namespace arch
|
||||
} // namespace cutlass
|
||||
|
||||
Reference in New Issue
Block a user