diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/test_ops.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/test_ops.cc index 12390b6a317d25b01eee26e0d1c42bc028b5a150..c0952440da74b420f4069c9523da5b677d4da037 100644 --- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/test_ops.cc +++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/test_ops.cc @@ -572,6 +572,35 @@ void testPerforation(){ +void testSampling(){ + + printf("***** Testing Perforation ***** \n\n"); + Tensor* input = (Tensor*) create4DTensor(CUDNN_DATA_FLOAT, CUDNN_TENSOR_NCHW, 1, 3, 4, 4); + fillTensorWithVal(input, 1); + //fillWithOnesAndTwos(input); + + Tensor* filter = (Tensor*) create4DTensor(CUDNN_DATA_FLOAT, CUDNN_TENSOR_NCHW, 1, 3, 3, 3); + fillTensorWithVal(filter, 1); + + float* host_ptr = (float*) ((struct Tensor*) input)->host_data; + //host_ptr[0] = 0; + //host_ptr[1] = 0; + //host_ptr[2] = 0; + //host_ptr[3] = 0; + //host_ptr[4] = 0; + //host_ptr[5] = 0; + //printTensorValues(input); + + void* res = tensorConvApprox(input, filter, 0, 0, 1, 1, 1, 1, 1, 1, 2, 1); + + printTensorValues(res); + +} + + + + + int main(){ @@ -602,7 +631,11 @@ int main(){ //testSampleFilter(); - testPerforation(); + //--- testPerforation(); + + + testSampling(); + stopProfiling();