Skip to content
Snippets Groups Projects
Commit 286015c6 authored by Hashim Sharif's avatar Hashim Sharif
Browse files

Removing erroneous prints from tensor-rt

parent 3a301edc
No related branches found
No related tags found
No related merge requests found
......@@ -333,12 +333,10 @@ void *wrapper_ConvLayer2(
// NOTE: out_min, out_max are only relevant for ClippedRelu
float out_min, float out_max) {
INFO("*** ------Conv Layer \n");
//INFO("*** ------Conv Layer \n");
NodeConfiguration *NodeConf = RC->getNodeConfiguration(hpvm_node_id);
INFO("HERE\n");
if (NodeConf->isGPUNodeConfiguration()) {
INFO("GPU Configuration for ConvLayer\n");
// Mapped to GPU - get a GPU node configuration
GPUNodeConfiguration *GPUConf = (GPUNodeConfiguration *)NodeConf;
......@@ -461,8 +459,7 @@ void *wrapper_ConvLayer2(
}
return pool_out;
} else if (NodeConf->isCPUNodeConfiguration()) {
INFO("CPU Configuration for ConvLayer\n");
// Mapped to CPU - get a CPU node configuration
// Mapped to CPU - get a CPU node configuration
CPUNodeConfiguration *CPUConf = (CPUNodeConfiguration *)NodeConf;
std::vector< std::pair< CPUNodeConfiguration::TENSOR_OP,
......@@ -495,7 +492,6 @@ void *wrapper_ConvLayer2(
switch (activation_id) {
case -1:
{ // No activation
INFO("No activation Function\n");
activation_out = add_out;
}
break;
......@@ -603,8 +599,6 @@ wrapper_FCLayer(const char *hpvm_node_id, void *input, void *weights,
// NOTE: out_min and out_max are only relevant for ClippedRelu
float out_min, float out_max) {
INFO("*** Dense Layer \n");
NodeConfiguration *NodeConf = RC->getNodeConfiguration(hpvm_node_id);
if (NodeConf->isGPUNodeConfiguration()) {
DEBUG("GPU Configuration for FCLayer\n");
......@@ -696,7 +690,6 @@ wrapper_FCLayer(const char *hpvm_node_id, void *input, void *weights,
{ // No activation
CUSTOM_ASSERT((ApproxChoices.size() == 2) &&
"Incorrect number of operations in provided FC layer configuration");
INFO("No activation Function\n");
activation_out = add_out;
}
break;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment