From bfc30fa6ea77f29ce2439b8e4fd7c4f45217c559 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 14 May 2025 19:34:52 -0700 Subject: [PATCH] hotfix: typo in shm_name --- examples/mlperf/dataloader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/mlperf/dataloader.py b/examples/mlperf/dataloader.py index b513ccbc08..0942e83615 100644 --- a/examples/mlperf/dataloader.py +++ b/examples/mlperf/dataloader.py @@ -129,7 +129,7 @@ def batch_load_resnet(batch_size=64, val=False, shuffle=True, seed=None, pad_fir q_in, q_out = Queue(), Queue() sz = (batch_size*BATCH_COUNT, 224, 224, 3) - shm_name = "resnet_X_val" if val else "resnet_X_train", + shm_name = "resnet_X_val" if val else "resnet_X_train" if not OSX and os.path.exists(f"/dev/shm/{shm_name}"): os.unlink(f"/dev/shm/{shm_name}") shm = shared_memory.SharedMemory(name=shm_name, create=True, size=prod(sz)) procs = []