-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrai_build.yml
30 lines (30 loc) · 1.51 KB
/
rai_build.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
rai:
version: 0.2
image: illinoisimpact/ece408_mxnet_docker:amd64-gpu-latest
resources:
cpu:
architecture: amd64
gpu:
architecture: pascal
count: 1
network: false
commands:
build:
- /bin/bash -c "cp -rv /ece408_src/* /mxnet/src/operator/custom" # copy golden files to mxnet source tree
- /bin/bash -c "cp -rv /src/* /build" # copy the project folder to /build so everything appears in the upload
- /bin/bash -c "for src in ece408_src/*; do cp -v $src /mxnet/src/operator/custom/.; done" # copy source files to mxnet tree
- nice -n20 make -C /mxnet # build mxnet
- pip2 install --user -e /mxnet/python # install python bindings
# - /usr/bin/time python m1.1.py # execute code
#- /usr/bin/time python m1.2.py # execute code
# - nvprof python m1.2.py
# /usr/bin/time python m2.1.py 10000 # execute code
- /usr/bin/time python m3.1.py 100 # execute code
- /usr/bin/time python m3.1.py 1000 # execute code
- /usr/bin/time python m3.1.py # execute code
#- /usr/bin/time python m4.1.py # execute code
# - nvprof -o timeline.nvvp python m3.1.py 10000 # execute code
# - nvprof --kernels "::forward:1" --analysis-metrics -o forward1_analysis.nvvp python m3.1.py 10000 # execute code
# - nvprof --kernels "::forward:2" --analysis-metrics -o forward2_analysis.nvvp python m3.1.py 10000 # execute code
- nvprof -o timeline.nvvp python m3.1.py 100 # execute code
- nvprof --analysis-metrics -o forward1_analysis.nvvp python m3.1.py 100 # execute code