30 #if !defined(__CUDACC_RTC__) 39 template<
typename DeviceGemmTraits_ >
44 typedef typename Traits::Params
Params;
47 #if !defined(__CUDACC_RTC__) 48 static __host__ cudaError_t
launch(
Params const& params) {
50 Traits::GemmTraits::KernelClass::launch(params.GemmParams);
51 cudaError_t err = cudaGetLastError();
52 if (err != cudaSuccess)
54 Traits::ReductionTraits::KernelClass::launch(params.ReductionParams);
55 return cudaGetLastError();
DeviceGemmTraits_ Traits
The Traits.
Definition: device_gemm.h:42
A Coord is a coordinate of arbitrary rank into a tensor or matrix.
Definition: device_gemm.h:40
Traits::Params Params
Use the params object defined in traits.
Definition: device_gemm.h:44
CUTLASS_DEVICE DeviceGemm()
Ctor.
Definition: device_gemm.h:64
static __host__ cudaError_t launch(Params const ¶ms)
Support for NVRTC.
Definition: device_gemm.h:49