Skip to content

Commit 447d628

Browse files
authored
Merge pull request #1086 from trz42/2023.06-a64fx-2023a-eb491-apps-pytorch
{2023.06}[2023a,a64fx] PyTorch 2.1.2 originally built with EB 4.9.0
2 parents 7a579c3 + cfef1ab commit 447d628

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

eb_hooks.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1286,6 +1286,9 @@ def set_maximum(parallel, max_value):
12861286
'MBX': {
12871287
'*': (divide_by_factor, 2),
12881288
},
1289+
'PyTorch': {
1290+
CPU_TARGET_A64FX: (divide_by_factor, 4),
1291+
},
12891292
'TensorFlow': {
12901293
'*': (divide_by_factor, 2),
12911294
CPU_TARGET_A64FX: (set_maximum, 8),

0 commit comments

Comments
 (0)