File tree Expand file tree Collapse file tree 12 files changed +1
-366
lines changed Expand file tree Collapse file tree 12 files changed +1
-366
lines changed Original file line number Diff line number Diff line change @@ -233,17 +233,6 @@ else
233
233
make VERBOSE=1
234
234
popd
235
235
assert_git_not_dirty
236
-
237
- # Build custom backend tests.
238
- CUSTOM_BACKEND_BUILD=" $PWD /../custom-backend-build"
239
- CUSTOM_BACKEND_TEST=" $PWD /test/custom_backend"
240
- python --version
241
- mkdir " $CUSTOM_BACKEND_BUILD "
242
- pushd " $CUSTOM_BACKEND_BUILD "
243
- cmake " $CUSTOM_BACKEND_TEST " -DCMAKE_PREFIX_PATH=" $SITE_PACKAGES /torch" -DPYTHON_EXECUTABLE=" $( which python) "
244
- make VERBOSE=1
245
- popd
246
- assert_git_not_dirty
247
236
else
248
237
# Test standalone c10 build
249
238
if [[ " $BUILD_ENVIRONMENT " == * xenial-cuda10.1-cudnn7-py3* ]]; then
Original file line number Diff line number Diff line change @@ -99,26 +99,6 @@ test_libtorch() {
99
99
fi
100
100
}
101
101
102
- test_custom_backend () {
103
- echo " Testing custom backends"
104
- pushd test/custom_backend
105
- rm -rf build && mkdir build
106
- pushd build
107
- SITE_PACKAGES=" $( python -c ' from distutils.sysconfig import get_python_lib; print(get_python_lib())' ) "
108
- CMAKE_PREFIX_PATH=" $SITE_PACKAGES /torch" cmake ..
109
- make VERBOSE=1
110
- popd
111
-
112
- # Run Python tests and export a lowered module.
113
- python test_custom_backend.py -v
114
- python backend.py --export-module-to=model.pt
115
- # Run C++ tests using the exported module.
116
- build/test_custom_backend ./model.pt
117
- rm -f ./model.pt
118
- popd
119
- assert_git_not_dirty
120
- }
121
-
122
102
test_custom_script_ops () {
123
103
echo " Testing custom script operators"
124
104
pushd test/custom_operator
@@ -144,13 +124,11 @@ if [ -z "${BUILD_ENVIRONMENT}" ] || [[ "${BUILD_ENVIRONMENT}" == *-test ]]; then
144
124
test_python_all
145
125
test_libtorch
146
126
test_custom_script_ops
147
- test_custom_backend
148
127
else
149
128
if [[ " ${BUILD_ENVIRONMENT} " == * -test1 ]]; then
150
129
test_python_all
151
130
elif [[ " ${BUILD_ENVIRONMENT} " == * -test2 ]]; then
152
131
test_libtorch
153
132
test_custom_script_ops
154
- test_custom_backend
155
133
fi
156
134
fi
Original file line number Diff line number Diff line change @@ -228,23 +228,6 @@ test_libtorch() {
228
228
fi
229
229
}
230
230
231
- test_custom_backend () {
232
- if [[ " $BUILD_ENVIRONMENT " != * rocm* ]] && [[ " $BUILD_ENVIRONMENT " != * asan* ]] ; then
233
- echo " Testing custom backends"
234
- CUSTOM_BACKEND_BUILD=" $PWD /../custom-backend-build"
235
- pushd test/custom_backend
236
- cp -a " $CUSTOM_BACKEND_BUILD " build
237
- # Run tests Python-side and export a lowered module.
238
- python test_custom_backend.py -v
239
- python backend.py --export-module-to=model.pt
240
- # Run tests C++-side and load the exported lowered module.
241
- build/test_custom_backend ./model.pt
242
- rm -f ./model.pt
243
- popd
244
- assert_git_not_dirty
245
- fi
246
- }
247
-
248
231
test_custom_script_ops () {
249
232
if [[ " $BUILD_ENVIRONMENT " != * rocm* ]] && [[ " $BUILD_ENVIRONMENT " != * asan* ]] ; then
250
233
echo " Testing custom script operators"
@@ -348,7 +331,6 @@ elif [[ "${BUILD_ENVIRONMENT}" == *-test2 || "${JOB_BASE_NAME}" == *-test2 ]]; t
348
331
test_aten
349
332
test_libtorch
350
333
test_custom_script_ops
351
- test_custom_backend
352
334
test_torch_function_benchmark
353
335
elif [[ " ${BUILD_ENVIRONMENT} " == * -bazel-* ]]; then
354
336
test_bazel
364
346
test_aten
365
347
test_libtorch
366
348
test_custom_script_ops
367
- test_custom_backend
368
349
test_torch_function_benchmark
369
350
fi
Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -41,15 +41,13 @@ run_tests() {
41
41
$SCRIPT_HELPERS_DIR /test_python_nn.bat " $DETERMINE_FROM " && \
42
42
$SCRIPT_HELPERS_DIR /test_python_all_except_nn.bat " $DETERMINE_FROM " && \
43
43
$SCRIPT_HELPERS_DIR /test_custom_script_ops.bat && \
44
- $SCRIPT_HELPERS_DIR /test_custom_backend.bat && \
45
44
$SCRIPT_HELPERS_DIR /test_libtorch.bat
46
45
else
47
46
if [[ " ${JOB_BASE_NAME} " == * -test1 ]]; then
48
47
$SCRIPT_HELPERS_DIR /test_python_nn.bat " $DETERMINE_FROM " && \
49
48
$SCRIPT_HELPERS_DIR /test_libtorch.bat
50
49
elif [[ " ${JOB_BASE_NAME} " == * -test2 ]]; then
51
50
$SCRIPT_HELPERS_DIR /test_python_all_except_nn.bat " $DETERMINE_FROM " && \
52
- $SCRIPT_HELPERS_DIR /test_custom_backend.bat && \
53
51
$SCRIPT_HELPERS_DIR /test_custom_script_ops.bat
54
52
fi
55
53
fi
Original file line number Diff line number Diff line change
1
+ #include < torch/csrc/jit/api/module.h>
1
2
#include < torch/csrc/jit/backends/backend.h>
2
3
3
4
namespace torch {
Load Diff This file was deleted.
Load Diff This file was deleted.
Load Diff This file was deleted.
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments