Skip to content

Commit fbb3f8e

Browse files
fix: libtorch error when building perception
Change-Id: I7caa205e88c105e13d2c33f9fb3436b29251e994
1 parent 1c36409 commit fbb3f8e

File tree

4 files changed

+2
-21
lines changed

4 files changed

+2
-21
lines changed

modules/perception/common/inference/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ apollo_cc_library(
2626
hdrs = ["inference.h"],
2727
deps = [
2828
"//modules/perception/common/base:apollo_perception_common_base",
29-
"//third_party:libtorch",
29+
"@libtorch_gpu",
3030
"@boost",
3131
],
3232
)

modules/perception/common/inference/inference_factory.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
#include "modules/perception/common/inference/onnx/libtorch_obstacle_detector.h"
2121
#include "modules/perception/common/inference/onnx/onnx_single_batch_infer.h"
2222
#include "modules/perception/common/inference/paddlepaddle/paddle_net.h"
23-
#include "modules/perception/common/inference/tensorrt/rt_net.h"
2423
#if GPU_PLATFORM == NVIDIA
2524
#include "modules/perception/common/inference/tensorrt/rt_net.h"
2625
#define RTNET RTNet(proto_file, weight_file, outputs, inputs)

third_party/BUILD

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +0,0 @@
1-
load("@rules_cc//cc:defs.bzl", "cc_library")
2-
3-
package(default_visibility = ["//visibility:public"])
4-
5-
cc_library(
6-
name = "libtorch",
7-
deps = select({
8-
"//tools/platform:use_nvidia_gpu": [
9-
"@libtorch_gpu",
10-
],
11-
"//tools/platform:use_amd_gpu": [
12-
"@libtorch_gpu",
13-
],
14-
"//conditions:default": [
15-
"@libtorch_cpu",
16-
],
17-
}),
18-
)

third_party/libtorch/workspace.bzl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def repo_gpu():
1818
build_file = select({
1919
"@local_config_rocm//rocm:using_hipcc": clean_dep("//third_party/libtorch:libtorch_gpu_rocm.BUILD"),
2020
"//conditions:default": clean_dep("//third_party/libtorch:libtorch_gpu_cuda.BUILD"),
21-
})
21+
}),
2222
path = "/usr/local/libtorch_gpu/include",
2323
)
2424
# native.new_local_repository(

0 commit comments

Comments
 (0)