Commit 4a20c076 authored by Astha Rai's avatar Astha Rai
Browse files

updated test/profiler files

parent baaad9ec
...@@ -43,14 +43,20 @@ bool profile_transpose_impl(int do_verification, ...@@ -43,14 +43,20 @@ bool profile_transpose_impl(int do_verification,
int init_method, int init_method,
bool do_log, bool do_log,
bool time_kernel, bool time_kernel,
ck::index_t N, index_t N,
ck::index_t C, index_t C,
ck::index_t D, index_t D,
ck::index_t H, index_t H,
ck::index_t W) index_t W)
{ {
bool pass = true; bool pass = true;
/**index_t N = lengths[0];
index_t C = lengths[1];
index_t D = lengths[2];
index_t H = lengths[3];
index_t W = lengths[4];**/
std::vector<ck::index_t> ncdhw = {N, C, D, H, W}; std::vector<ck::index_t> ncdhw = {N, C, D, H, W};
std::vector<ck::index_t> ndhwc = {N, D, H, W, C}; std::vector<ck::index_t> ndhwc = {N, D, H, W, C};
Tensor<ADataType> a(ncdhw); Tensor<ADataType> a(ncdhw);
......
...@@ -11,7 +11,7 @@ using F16 = ck::half_t; ...@@ -11,7 +11,7 @@ using F16 = ck::half_t;
using F32 = float; using F32 = float;
template <typename Tuple> template <typename Tuple>
class TestTranspose : public ck::test::TestTranspose<Tuple> class TestTranspose : public ::testing::Test
{ {
}; };
......
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