"tests/models/vscode:/vscode.git/clone" did not exist on "7d887118b9e954a265060b3d6cbd061f19ae315b"
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 @@ ...@@ -13,7 +13,7 @@
#include "ck/ck.hpp" #include "ck/ck.hpp"
#include "ck/tensor_operation/gpu/element/binary_element_wise_operation.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/utility/type.hpp"
#include "ck/library/utility/check_err.hpp" #include "ck/library/utility/check_err.hpp"
......
...@@ -7,7 +7,7 @@ using ADataType = F16; ...@@ -7,7 +7,7 @@ using ADataType = F16;
using BDataType = F16; using BDataType = F16;
using DeviceElementwisePermuteInstance = ck::tensor_operation::device:: 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" #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