Skip to content

Commit 2077b37

Browse files
PaliCfacebook-github-bot
authored andcommitted
Remove TARGETS for caffe2/torch/csrc/deploy and remove stragglers (#150)
Summary: Pull Request resolved: #150 This pr finishes off the migration to `multipy/runtime` from `torch/csrc/deploy`. Reviewed By: d4l3k Differential Revision: D39174555 fbshipit-source-id: 311a010208bddbe4f0995c1dc166ab71397578af
1 parent 45e6401 commit 2077b37

File tree

3 files changed

+7
-8
lines changed

3 files changed

+7
-8
lines changed

multipy/runtime/benchmark.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ int main(int argc, char* argv[]) {
313313
// make sure gpu_wrapper.py is in the import path
314314
for (auto& interp : manager.allInstances()) {
315315
auto I = interp.acquireSession();
316-
I.global("sys", "path").attr("append")({"torch/csrc/deploy/example"});
316+
I.global("sys", "path").attr("append")({"multipy/runtime/example"});
317317
}
318318

319319
auto n_threads = {1, 2, 4, 8, 16, 32, 40};

multipy/runtime/example/benchmark.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ int main(int argc, char* argv[]) {
313313
// make sure gpu_wrapper.py is in the import path
314314
for (auto& interp : manager.allInstances()) {
315315
auto I = interp.acquireSession();
316-
I.global("sys", "path").attr("append")({"torch/csrc/deploy/example"});
316+
I.global("sys", "path").attr("append")({"multipy/runtime/example"});
317317
}
318318

319319
auto n_threads = {1, 2, 4, 8, 16, 32, 40};

multipy/runtime/test_deploy_gpu.cpp

+5-6
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ int main(int argc, char* argv[]) {
1919
return rc;
2020
}
2121

22-
const char* simple = "torch/csrc/deploy/example/generated/simple";
23-
const char* simple_jit = "torch/csrc/deploy/example/generated/simple_jit";
22+
const char* simple = "multipy/runtime/example/generated/simple";
23+
const char* simple_jit = "multipy/runtime/example/generated/simple_jit";
2424

2525
const char* path(const char* envname, const char* path) {
2626
const char* e = getenv(envname);
@@ -63,8 +63,7 @@ TEST(TorchDeployGPUTest, SimpleModel) {
6363

6464
TEST(TorchDeployGPUTest, UsesDistributed) {
6565
const auto model_filename = path(
66-
"USES_DISTRIBUTED",
67-
"torch/csrc/deploy/example/generated/uses_distributed");
66+
"USES_DISTRIBUTED", "multipy/runtime/example/generated/uses_distributed");
6867
torch::deploy::InterpreterManager m(1);
6968
torch::deploy::Package p = m.loadPackage(model_filename);
7069
{
@@ -81,7 +80,7 @@ TEST(TorchDeployGPUTest, UsesCuda) {
8180
#endif
8281

8382
const auto model_filename =
84-
path("USES_CUDA", "torch/csrc/deploy/example/generated/uses_cuda");
83+
path("USES_CUDA", "multipy/runtime/example/generated/uses_cuda");
8584
torch::deploy::InterpreterManager m(1);
8685
torch::deploy::Package p = m.loadPackage(model_filename);
8786
{
@@ -96,7 +95,7 @@ TEST(TorchDeployGPUTest, TensorRT) {
9695
GTEST_SKIP();
9796
}
9897
auto packagePath = path(
99-
"MAKE_TRT_MODULE", "torch/csrc/deploy/example/generated/make_trt_module");
98+
"MAKE_TRT_MODULE", "multipy/runtime/example/generated/make_trt_module");
10099
torch::deploy::InterpreterManager m(1);
101100
torch::deploy::Package p = m.loadPackage(packagePath);
102101
auto makeModel = p.loadPickle("make_trt_module", "model.pkl");

0 commit comments

Comments
 (0)