Skip to content

Commit 0e4dfbd

Browse files
committed
gpu: add corona test examples
Signed-off-by: vsoch <[email protected]>
1 parent 796c04b commit 0e4dfbd

File tree

7 files changed

+27
-2
lines changed

7 files changed

+27
-2
lines changed
79.5 KB
Loading
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# Assigns each rank to its GPU and a dedicated L2 cache (and thus, a dedicated core)
2+
# within that GPU's local NUMA domain.
3+
default:
4+
type: core
5+
bind: gpu-local
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# Assigns each rank to its GPU and a dedicated L2 cache (and thus, a dedicated core)
2+
# within that GPU's local NUMA domain.
3+
default:
4+
type: core
5+
bind: gpu-remote
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# Assigns each rank to its GPU and a dedicated L2 cache (and thus, a dedicated core)
2+
# within that GPU's local NUMA domain.
3+
default:
4+
type: l2cache
5+
bind: gpu-local
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# Assigns each rank to its GPU and a dedicated L2 cache (and thus, a dedicated core)
2+
# within that GPU's local NUMA domain.
3+
default:
4+
type: l2cache
5+
bind: gpu-remote
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# Assigns each rank to its GPU and a dedicated L2 cache (and thus, a dedicated core)
2+
# within that GPU's local NUMA domain.
3+
default:
4+
type: l3cache
5+
bind: gpu-remote

fluxbind/shape/shape.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,11 +182,11 @@ def get_binding_in_gpu_domain(
182182
if not hwloc_type:
183183
raise ValueError("Rule with GPU binding must have a 'type'.")
184184

185-
if hwloc_type in ["numa", "package", "l3cache", "machine"]:
185+
if hwloc_type in ["numa", "package", "machine"]:
186186
# If a broad type is requested, the binding is the domain itself.
187187
return domain
188188

189-
elif hwloc_type in ["core", "pu", "l2cache"]:
189+
elif hwloc_type in ["core", "pu", "l2cache", "l3cache"]:
190190

191191
# Get the number of objects to select, defaulting to 1.
192192
count = rule.get("count", 1)

0 commit comments

Comments
 (0)