summaryrefslogtreecommitdiff
path: root/app/relay/interpreters.js
diff options
context:
space:
mode:
Diffstat (limited to 'app/relay/interpreters.js')
-rw-r--r--app/relay/interpreters.js22
1 files changed, 22 insertions, 0 deletions
diff --git a/app/relay/interpreters.js b/app/relay/interpreters.js
new file mode 100644
index 0000000..1adb95e
--- /dev/null
+++ b/app/relay/interpreters.js
@@ -0,0 +1,22 @@
+export default {
+ pytorch: {
+ cmd: process.env.PYTORCH_BIN,
+ gpu: true,
+ },
+ tensorflow: {
+ cmd: process.env.TENSORFLOW_BIN,
+ gpu: true,
+ },
+ python: {
+ cmd: process.env.PYTHON_BIN,
+ gpu: false,
+ },
+ bash: {
+ cmd: process.env.BASH_BIN || '/bin/bash',
+ gpu: false,
+ }
+ perl: {
+ cmd: process.env.PERL_BIN || '/usr/bin/perl',
+ gpu: false,
+ }
+} \ No newline at end of file