Skip to content
Snippets Groups Projects
Commit 4d15aefd authored by kotsifa2's avatar kotsifa2
Browse files

Removed in place checks from pool in CUDNN backend

parent 03aa6c11
No related branches found
No related tags found
No related merge requests found
......@@ -426,15 +426,6 @@ void CGT_CUDNN::codeGen(DFLeafNode* N) {
case Intrinsic::visc_tensor_pool_mean:
{ /* llvm.visc.tensor.relu */
DEBUG(errs() << F_cudnn->getName() << "\t: Handling tensor_pool_max\n");
// Tensor relu(a) is in place for argument a.
Value *Op = II->getOperand(0);
// Test the intrinsic operand for in place operation.
bool inplace = isValidOperandForInPlaceOperation(Op, F_cudnn, N);
// Code generation cannot continue if this is false, because the target
// only provides an in place operation
assert(inplace &&
"Operand not valid for in place operation. Code gen aborted.\n");
// Argument list - tensorPooling(input, poolFunction, window_height, window_width, vertical_pad, horizontal_pad,
// vertical_stride, horizontal_stride);
......
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