changed vector load back, still causing errors

1a7cc199 · Astha Rai · b7d7d6d0 · 1a7cc199
Commit 1a7cc199 authored Oct 11, 2022 by Astha Rai
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

example/43_elementwise_permute_2d/elementwise_permute_4D_fp16_2d.cpp ...elementwise_permute_2d/elementwise_permute_4D_fp16_2d.cpp +2 -2

No files found.
--- a/example/43_elementwise_permute_2d/elementwise_permute_4D_fp16_2d.cpp
+++ b/example/43_elementwise_permute_2d/elementwise_permute_4D_fp16_2d.cpp
@@ -27,7 +27,7 @@ using DeviceElementwisePermuteInstance =
                                                    8,
                                                    8,
                                                    ck::Sequence<8>,
-                                                    ck::Sequence<1>>;
+                                                    ck::Sequence<8>>;

 template <typename HostTensorA, typename HostTensorB, typename Functor>
 void host_elementwise4D(HostTensorB& B_nhwc,
@@ -48,7 +48,7 @@ void host_elementwise4D(HostTensorB& B_nhwc,
 int main()
 {
    bool do_verification = true;
-    bool time_kernel     = false;
+    bool time_kernel     = true;

    std::vector<std::size_t> nchw = {4, 4, 8, 8};
    std::vector<std::size_t> nhwc = {4, 8, 8, 4};