14 lines
329 B
Bash
14 lines
329 B
Bash
|
#!/bin/bash -e
|
||
|
|
echo "********* CPU *********"
|
||
|
|
CPU=1 python3 "$@"
|
||
|
|
echo "********* GPU *********"
|
||
|
|
GPU=1 python3 "$@"
|
||
|
|
echo "********* METAL *********"
|
||
|
|
METAL=1 python3 "$@"
|
||
|
|
echo "********* CLANG *********"
|
||
|
|
CLANG=1 python3 "$@"
|
||
|
|
echo "********* LLVM *********"
|
||
|
|
LLVM=1 python3 "$@"
|
||
|
|
echo "********* TORCH *********"
|
||
|
|
TORCH=1 python3 "$@"
|