device_contraction_utils.hpp File Reference

device_contraction_utils.hpp File Reference#

Composable Kernel: device_contraction_utils.hpp File Reference
device_contraction_utils.hpp File Reference
#include <cassert>
#include <sstream>
#include <vector>
#include "ck/ck.hpp"

Go to the source code of this file.

Namespaces

namespace  ck
namespace  ck::tensor_operation
namespace  ck::tensor_operation::device

Functions

template<index_t NumDim1, index_t NumDim2>
auto ck::tensor_operation::device::CalculateMaxRead (const std::vector< index_t > &lengths, const std::vector< index_t > &strides)