Skip to content
Permalink

Comparing changes

This is a direct comparison between two commits made in this repository or its related repositories. View the default comparison for this range or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: InfiniTensor/RefactorGraph
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: 9f34dda3e40d889fac869c71c03fb89c135b0905
Choose a base ref
..
head repository: InfiniTensor/RefactorGraph
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: 385ce12f9738407bd62368cab39a0500850cf264
Choose a head ref
Showing with 6 additions and 0 deletions.
  1. +6 −0 src/04kernel/src/collectors/pool.cc
6 changes: 6 additions & 0 deletions src/04kernel/src/collectors/pool.cc
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
#include "kernel/collectors/pool.h"
#include "../kernels/pool/cudnn_kernel.hh"
#include "../kernels/pool/cnnl_kernel.hh"

namespace refactor::kernel {

@@ -29,6 +30,11 @@ namespace refactor::kernel {
ans.emplace_back(std::move(ptr));
}
break;
case decltype(_target)::Mlu:
if (auto ptr = PoolCnnl::build(type, ceil, kernelShape, attributes, x, y); ptr) {
ans.emplace_back(std::move(ptr));
}
break;
default:
UNREACHABLEX(void, "Unknown target");
}