search

LeelaChessZero / lc0

The rewritten engine, originally for tensorflow. Now all other backends have been ported here.
GNU General Public License v3.0
2.44k stars 529 forks source link

Use sliceTensor instead of splitTensor for attn policy promo #1828

Closed almaudoh closed 1 year ago

almaudoh commented 1 year ago

Versions of macos prior to 12.3 do not have [MPSGraph splitTensor]. This PR changes attn policy promo computation to use sliceTensor to support older macos 12.x versions.

almaudoh commented 1 year ago 
using splitTensor
===========================
Total time (ms) : 381612
Nodes searched  : 39545
Nodes/second    : 104
using sliceTensor
===========================
Total time (ms) : 383364
Nodes searched  : 40287
Nodes/second    : 105