Commit 9c5dd6bf authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Use 'DevicePermute' device op in example

parent 1fdcf492
......@@ -13,7 +13,7 @@
#include "ck/ck.hpp"
#include "ck/tensor_operation/gpu/element/binary_element_wise_operation.hpp"
#include "ck/tensor_operation/gpu/device/device_elementwise.hpp"
#include "ck/tensor_operation/gpu/device/device_permute.hpp"
#include "ck/utility/type.hpp"
#include "ck/library/utility/check_err.hpp"
......
......@@ -7,7 +7,7 @@ using ADataType = F16;
using BDataType = F16;
using DeviceElementwisePermuteInstance = ck::tensor_operation::device::
DeviceElementwise<ck::Tuple<ADataType>, ck::Tuple<BDataType>, PassThrough, 4, 8, S<8>, S<1>>;
DevicePermute<ck::Tuple<ADataType>, ck::Tuple<BDataType>, PassThrough, 4, 8, S<8>, S<1>>;
#include "run_elementwise_permute_example.inc"
......
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