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