Commit bf523dbe authored by turneram's avatar turneram
Browse files

Formatting

parent 44a12304
......@@ -50,7 +50,7 @@ constexpr auto MakeDescriptor_M(const L& lengths, const S& strides, const N& /*
auto gridSize = 72;
auto blockSize = 1024;
constexpr auto ndim = 1;
//auto idx = make_index();
// auto idx = make_index();
auto tupleOfShape = generate_tuple([&](auto I) { return static_cast<ck::index_t>(lengths[I]); },
ck::Number<ndim>{});
auto tupleOfStride = generate_tuple(
......@@ -68,7 +68,7 @@ constexpr auto MakeDescriptor_M(const L& lengths, const S& strides, const N& /*
}
const auto M = desc_m.GetLength(I0);
const ck::index_t loop_step = /* idx.nglobal(); // */ gridSize * blockSize/* * MPerThread */;
const ck::index_t loop_step = /* idx.nglobal(); // */ gridSize * blockSize /* * MPerThread */;
const auto pad = ck::math::integer_least_multiple(M, loop_step) - M;
const auto desc_m_pad =
transform_tensor_descriptor(desc_m,
......@@ -91,7 +91,7 @@ template <class T, class U, class V>
__device__ void ck_elementwise(const T& a_t, const U& b_t, const V& c_t)
{
auto idx = make_index();
if (idx.global == 0)
if(idx.global == 0)
{
constexpr auto lengths = get_shape_c<T>{}.lens;
constexpr auto strides = get_shape_c<T>{}.strides;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment