lbann icon indicating copy to clipboard operation
lbann copied to clipboard
verified publisher icon LLNL

Adds implementation of distconv enabled 3D MatMul

Open szaman19 opened this issue 3 years ago • 0 comments

Enables splitting along the 1st dimension in a 4D tensor and using distconv to perform StridedBatchedGEMMs.

  • Need to resolve issue #2126 prior to merging

szaman19 avatar Aug 06 '22 04:08 szaman19