Commit 949de23b authored by Astha Rai's avatar Astha Rai
Browse files

example with 5D input tensor and 2d kernel - not working: issues with arguments

parent b472cdf6
...@@ -48,11 +48,14 @@ int main() ...@@ -48,11 +48,14 @@ int main()
bool do_verification = true; bool do_verification = true;
bool time_kernel = true; bool time_kernel = true;
const int N = 120; //const int N = 120;
const int C = 128; //const int C = 128;
const int H = 32; //const int H = 32;
const int W = 1024; //const int W = 1024;
const int N = 16;
const int C = 8;
const int H = 8;
const int W = 8;
/**const int N = 120; /**const int N = 120;
const int H = 32; const int H = 32;
const int W = 64; const int W = 64;
...@@ -107,6 +110,8 @@ int main() ...@@ -107,6 +110,8 @@ int main()
float gb_per_sec = num_btype / 1.E6 / ave_time; float gb_per_sec = num_btype / 1.E6 / ave_time;
LogRangeAsType<float>(std::cout << "A : ", a.mData, ",") << std::endl;
//LogRangeAsType<float>(std::cout << "B : ", b.mData, ",") << std::endl;
std::cout << "Perf: " << ave_time << " ms, " << tflops << " TFlops, " << gb_per_sec << " GB/s" std::cout << "Perf: " << ave_time << " ms, " << tflops << " TFlops, " << gb_per_sec << " GB/s"
<< std::endl; << std::endl;
...@@ -120,6 +125,7 @@ int main() ...@@ -120,6 +125,7 @@ int main()
Tensor<BDataType> host_b(nhwc); Tensor<BDataType> host_b(nhwc);
host_elementwise4D<Tensor<ADataType>, Tensor<BDataType>, PassThrough>( host_elementwise4D<Tensor<ADataType>, Tensor<BDataType>, PassThrough>(
host_b, a, nchw, PassThrough{}); host_b, a, nchw, PassThrough{});
//LogRangeAsType<float>(std::cout << "Host_b : ", host_b.mData, ",") << std::endl;
// LogRangeAsType<float>(std::cout << "Host b : ", host_b.mData, ",") << std::endl; // LogRangeAsType<float>(std::cout << "Host b : ", host_b.mData, ",") << std::endl;
pass &= pass &=
......
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