Skip to content

Commit

Permalink
fix benchmark test example manifest
Browse files Browse the repository at this point in the history
fixes #52
  • Loading branch information
arno01 authored and y2kenny-amd committed Jan 27, 2024
1 parent a763729 commit f27abcb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion example/pod/alexnet-gpu.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ spec:
- name: HIP_VISIBLE_DEVICES
value: "0" # # 0,1,2,...,n for running on GPU and select the GPUs, -1 for running on CPU
command: ["/bin/bash", "-c", "--"]
args: ["python3 benchmarks/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --model=alexnet; trap : TERM INT; sleep infinity & wait"]
args: ["python3 /benchmarks/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --model=alexnet; trap : TERM INT; sleep infinity & wait"]
resources:
limits:
amd.com/gpu: 1 # requesting a GPU

0 comments on commit f27abcb

Please sign in to comment.