"...composable_kernel.git" did not exist on "30072aec376a2fc6c6a465fe632bc7be050ec5ea"
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 ...@@ -37,8 +37,6 @@ struct SimpleDeviceMem
int main() int main()
{ {
bool time_kernel = true;
const int N = 16; const int N = 16;
const int C = 8; const int C = 8;
const int D = 8; const int D = 8;
...@@ -47,12 +45,12 @@ int main() ...@@ -47,12 +45,12 @@ int main()
std::vector<std::size_t> ncdhw = {N, C, D, H, W}; std::vector<std::size_t> ncdhw = {N, C, D, H, W};
std::vector<std::size_t> nchwd = {N, C, H, W, D}; std::vector<std::size_t> nchwd = {N, C, H, W, D};
Tensor<ADataType> a(ncdhw); // Tensor<ADataType> a(ncdhw);
Tensor<BDataType> b(nchwd); // Tensor<BDataType> b(nchwd);
auto size = N * C * D * H * W; 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> 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}; 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< ...@@ -37,7 +37,7 @@ struct DeviceOperationInstanceFactory<
DeviceElementwise<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>> DeviceElementwise<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>>
{ {
using DeviceOp = using DeviceOp =
DeviceElementwise3dImpl<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>; DeviceElementwise<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>;
static auto GetInstances() static auto GetInstances()
{ {
...@@ -52,7 +52,6 @@ struct DeviceOperationInstanceFactory< ...@@ -52,7 +52,6 @@ struct DeviceOperationInstanceFactory<
} }
} }
return op_ptrs; return op_ptrs;
}
}; };
} // namespace instance } // 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