Skip to content

Commit b144d06

Browse files
committed
added better print statement
1 parent 6dfd23c commit b144d06

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

synapse_net/inference/util.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -436,8 +436,6 @@ def get_default_tiling(is_2d: bool = False) -> Dict[str, Dict[str, int]]:
436436
tiling = {"tile": tile, "halo": halo}
437437

438438
elif torch.backends.mps.is_available(): # Check for Apple Silicon (MPS)
439-
print("Using Metal Performance Shaders (MPS) for inference.")
440-
# MPS memory detection is limited, so use conservative tiling
441439
tile = {"x": 512, "y": 512, "z": 64}
442440
halo = {"x": 64, "y": 64, "z": 16}
443441
print(f"Determined tile size for MPS: {tile}")
@@ -447,7 +445,7 @@ def get_default_tiling(is_2d: bool = False) -> Dict[str, Dict[str, int]]:
447445
# I am not sure what is reasonable on a cpu. For now choosing very small tiling.
448446
# (This will not work well on a CPU in any case.)
449447
else:
450-
print("Determining default tiling for CPU")
448+
print(f"Determining default tiling for CPU: {tile}")
451449
tiling = {
452450
"tile": {"x": 96, "y": 96, "z": 16},
453451
"halo": {"x": 16, "y": 16, "z": 4},

0 commit comments

Comments
 (0)