Skip to content

Commit 369dfb3

Browse files
committed
move contrib/inference to paddle/fluid/inference/api
1 parent 866fcb0 commit 369dfb3

26 files changed

+28
-57
lines changed

paddle/contrib/CMakeLists.txt

Lines changed: 0 additions & 16 deletions
This file was deleted.
File renamed without changes.

paddle/contrib/inference/paddle_inference_api_anakin_engine.cc renamed to paddle/fluid/inference/api/api_anakin_engine.cc

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
#include "paddle/contrib/inference/paddle_inference_api_anakin_engine.h"
1615
#include <cuda.h>
16+
#include "paddle/contrib/inference/paddle_inference_api_anakin_engine.h"
1717

1818
namespace paddle {
1919

@@ -47,8 +47,7 @@ bool PaddleInferenceAnakinPredictor::Run(
4747
}
4848
auto d_tensor_in_p = executor_.get_in(input.name);
4949
float *d_data_p = d_tensor_in_p->mutable_data();
50-
if (cudaMemcpy(d_data_p,
51-
static_cast<float *>(input.data.data()),
50+
if (cudaMemcpy(d_data_p, static_cast<float *>(input.data.data()),
5251
d_tensor_in_p->valid_size() * sizeof(float),
5352
cudaMemcpyHostToDevice) != 0) {
5453
LOG(ERROR) << "copy data from CPU to GPU error";
@@ -70,8 +69,7 @@ bool PaddleInferenceAnakinPredictor::Run(
7069
output.data.Resize(tensor->valid_size() * sizeof(float));
7170
}
7271
// Copy data from GPU -> CPU
73-
if (cudaMemcpy(output.data.data(),
74-
tensor->mutable_data(),
72+
if (cudaMemcpy(output.data.data(), tensor->mutable_data(),
7573
tensor->valid_size() * sizeof(float),
7674
cudaMemcpyDeviceToHost) != 0) {
7775
LOG(ERROR) << "copy data from GPU to CPU error";
@@ -106,9 +104,8 @@ std::unique_ptr<PaddlePredictor> PaddleInferenceAnakinPredictor::Clone() {
106104

107105
// A factory to help create difference predictor.
108106
template <>
109-
std::unique_ptr<PaddlePredictor>
110-
CreatePaddlePredictor<AnakinConfig, PaddleEngineKind::kAnakin>(
111-
const AnakinConfig &config) {
107+
std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
108+
AnakinConfig, PaddleEngineKind::kAnakin>(const AnakinConfig &config) {
112109
VLOG(3) << "Anakin Predictor create.";
113110
std::unique_ptr<PaddlePredictor> x(
114111
new PaddleInferenceAnakinPredictor(config));

paddle/contrib/inference/paddle_inference_api_anakin_engine.h renamed to paddle/fluid/inference/api/api_anakin_engine.h

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,7 @@ class PaddleInferenceAnakinPredictor : public PaddlePredictor {
4848
private:
4949
bool Init(const AnakinConfig& config);
5050

51-
anakin::graph::Graph<anakin::NV,
52-
anakin::saber::AK_FLOAT,
51+
anakin::graph::Graph<anakin::NV, anakin::saber::AK_FLOAT,
5352
anakin::Precision::FP32>
5453
graph_;
5554
anakin::Net<anakin::NV, anakin::saber::AK_FLOAT, anakin::Precision::FP32>

0 commit comments

Comments
 (0)