Commit 6b5bce42 authored by Astha Rai's avatar Astha Rai
Browse files

fixed error in instance generation

parent fb6efd98
......@@ -57,18 +57,8 @@ int main()
std::array<const void*, 1> input = {a_dev_buf.GetDeviceBuffer()};
std::array<void*, 1> output = {b_dev_buf.GetDeviceBuffer()};
using DeviceElementwisePermuteInstance =
ck::tensor_operation::device::DeviceElementwise3dImpl<ck::Tuple<ADataType>,
ck::Tuple<BDataType>,
PassThrough,
2,
2,
1,
8,
8,
8,
ck::Sequence<8>,
ck::Sequence<1>>;
using DeviceElementwisePermuteInstance = ck::tensor_operation::device::
DeviceElementwiseImpl<ck::Tuple<ADataType>, ck::Tuple<BDataType>, PassThrough, 5>;
// get device op instances
const auto op_ptrs = ck::tensor_operation::device::instance::DeviceOperationInstanceFactory<
......
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