Commit 1a7cc199 authored by Astha Rai's avatar Astha Rai
Browse files

changed vector load back, still causing errors

parent b7d7d6d0
...@@ -27,7 +27,7 @@ using DeviceElementwisePermuteInstance = ...@@ -27,7 +27,7 @@ using DeviceElementwisePermuteInstance =
8, 8,
8, 8,
ck::Sequence<8>, ck::Sequence<8>,
ck::Sequence<1>>; ck::Sequence<8>>;
template <typename HostTensorA, typename HostTensorB, typename Functor> template <typename HostTensorA, typename HostTensorB, typename Functor>
void host_elementwise4D(HostTensorB& B_nhwc, void host_elementwise4D(HostTensorB& B_nhwc,
...@@ -48,7 +48,7 @@ void host_elementwise4D(HostTensorB& B_nhwc, ...@@ -48,7 +48,7 @@ void host_elementwise4D(HostTensorB& B_nhwc,
int main() int main()
{ {
bool do_verification = true; 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> nchw = {4, 4, 8, 8};
std::vector<std::size_t> nhwc = {4, 8, 8, 4}; std::vector<std::size_t> nhwc = {4, 8, 8, 4};
......
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