| backend_dir=$(dirname $0) | |
| if [ -d $backend_dir/common ]; then | |
| source $backend_dir/common/libbackend.sh | |
| else | |
| source $backend_dir/../common/libbackend.sh | |
| fi | |
| if [ -d "/opt/intel" ]; then | |
| # Assumes we are using the Intel oneAPI container image | |
| # https://github.com/intel/intel-extension-for-pytorch/issues/538 | |
| export XPU=1 | |
| fi | |
| export PYTORCH_ENABLE_MPS_FALLBACK=1 | |
| startBackend $@ | |