#include <gridwise_gemm_xdlops_bwd_weight.hpp>
|
__host__ constexpr __device__ | Merge_v4_no_carry ()=default |
|
__host__ constexpr __device__ | Merge_v4_no_carry (const LowLengths &low_lengths) |
|
__host__ constexpr __device__ const auto & | GetUpperLengths () const |
|
template<typename LowIdx , typename UpIdx > |
__host__ constexpr __device__ void | CalculateLowerIndex (LowIdx &idx_low, const UpIdx &idx_up) const |
|
template<typename LowIdxDiff , typename UpIdxDiff , typename LowIdx , typename UpIdx , index_t Hack> |
__host__ __device__ void | UpdateLowerIndex (LowIdxDiff &idx_diff_low, const UpIdxDiff &idx_up_diff, LowIdx &idx_low, const UpIdx &idx_up_new, Number< Hack >) const |
|
__host__ __device__ void | Print () const |
|
◆ LowerIndex
template<typename LowLengths >
◆ LowLengthsScan
template<typename LowLengths >
◆ UpLengths
template<typename LowLengths >
◆ UpperIndex
template<typename LowLengths >
◆ Merge_v4_no_carry() [1/2]
template<typename LowLengths >
◆ Merge_v4_no_carry() [2/2]
template<typename LowLengths >
◆ CalculateLowerIndex()
template<typename LowLengths >
template<typename LowIdx , typename UpIdx >
__host__ constexpr __device__ void ck::Merge_v4_no_carry< LowLengths >::CalculateLowerIndex |
( |
LowIdx & |
idx_low, |
|
|
const UpIdx & |
idx_up |
|
) |
| const |
|
inlineconstexpr |
◆ GetNumOfLowerDimension()
template<typename LowLengths >
◆ GetNumOfUpperDimension()
template<typename LowLengths >
◆ GetUpperLengths()
template<typename LowLengths >
◆ IsKnownAtCompileTime()
template<typename LowLengths >
◆ IsLinearTransform()
template<typename LowLengths >
◆ IsValidUpperIndexAlwaysMappedToValidLowerIndex()
template<typename LowLengths >
__host__ static constexpr __device__ bool ck::Merge_v4_no_carry< LowLengths >::IsValidUpperIndexAlwaysMappedToValidLowerIndex |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ IsValidUpperIndexMappedToValidLowerIndex()
template<typename LowLengths >
template<typename UpIdx >
__host__ static constexpr __device__ bool ck::Merge_v4_no_carry< LowLengths >::IsValidUpperIndexMappedToValidLowerIndex |
( |
const UpIdx & |
| ) |
|
|
inlinestaticconstexpr |
◆ Print()
template<typename LowLengths >
◆ UpdateLowerIndex()
template<typename LowLengths >
template<typename LowIdxDiff , typename UpIdxDiff , typename LowIdx , typename UpIdx , index_t Hack>
__host__ __device__ void ck::Merge_v4_no_carry< LowLengths >::UpdateLowerIndex |
( |
LowIdxDiff & |
idx_diff_low, |
|
|
const UpIdxDiff & |
idx_up_diff, |
|
|
LowIdx & |
idx_low, |
|
|
const UpIdx & |
idx_up_new, |
|
|
Number< Hack > |
|
|
) |
| const |
|
inline |
◆ low_lengths_
template<typename LowLengths >
◆ low_lengths_scan_
template<typename LowLengths >
◆ NDimLow
template<typename LowLengths >
◆ up_lengths_
template<typename LowLengths >
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_bwd_weight.hpp