Skip to content

Commit 3f95cc6

Browse files
committed
Add --numactl_binding flag to host_device YAMLs
1 parent e5ac85e commit 3f95cc6

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

Ironwood/guides/automation/tpu7x-2x2x1-host_device.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ spec:
5353
pip install -r requirements.txt
5454
5555
GCS_BUCKET_DIR=${GCS_PATH}
56-
python Ironwood/src/run_benchmark.py --config="Ironwood/configs/host_device/host_device.yaml" --gcs-bucket-csv-dir=${GCS_BUCKET_DIR}
56+
python Ironwood/src/run_benchmark.py --config="Ironwood/configs/host_device/host_device.yaml" --gcs-bucket-csv-dir=${GCS_BUCKET_DIR} --numactl_binding
5757
resources:
5858
requests:
5959
google.com/tpu: 4

Ironwood/guides/host_device/tpu7x-host-device-benchmark.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ spec:
2424
cd accelerator-microbenchmarks
2525
pip install -r requirements.txt
2626
27-
bash ./Ironwood/scripts/run_host_device_benchmark.sh --config Ironwood/configs/host_device/host_device.yaml
27+
bash ./Ironwood/scripts/run_host_device_benchmark.sh --config Ironwood/configs/host_device/host_device.yaml --numactl_binding
2828
2929
resources:
3030
requests:

0 commit comments

Comments
 (0)