Commit 991ce41a authored by Astha Rai's avatar Astha Rai
Browse files

minor formatting and naming fixes

parent 09b0780d
......@@ -37,8 +37,6 @@ struct SimpleDeviceMem
int main()
{
bool time_kernel = true;
const int N = 16;
const int C = 8;
const int D = 8;
......@@ -47,12 +45,12 @@ int main()
std::vector<std::size_t> ncdhw = {N, C, D, H, W};
std::vector<std::size_t> nchwd = {N, C, H, W, D};
Tensor<ADataType> a(ncdhw);
Tensor<BDataType> b(nchwd);
// Tensor<ADataType> a(ncdhw);
// Tensor<BDataType> b(nchwd);
auto size = N * C * D * H * W;
a.GenerateTensorValue(GeneratorTensor_3<ADataType>{0.0, 1.0});
// a.GenerateTensorValue(GeneratorTensor_3<ADataType>{0.0, 1.0});
std::array<ck::index_t, 5> ab_lengths{N, C, H, W, D};
std::array<ck::index_t, 5> a_strides = {C * D * H * W, D * H * W, 1, D * H, D};
......
......@@ -37,7 +37,7 @@ struct DeviceOperationInstanceFactory<
DeviceElementwise<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>>
{
using DeviceOp =
DeviceElementwise3dImpl<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>;
DeviceElementwise<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>;
static auto GetInstances()
{
......@@ -52,7 +52,6 @@ struct DeviceOperationInstanceFactory<
}
}
return op_ptrs;
}
};
} // namespace instance
......
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