ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard
verified publisher icon hpcaitech

[tensor] added mlp implementation for the new sharding spec

Open FrankLeeeee opened this issue 3 years ago • 0 comments

The previous PR #1405 implemented the sharding spec. This PR implements the linear distributed computation using the new sharding spec API.

FrankLeeeee avatar Aug 09 '22 02:08 FrankLeeeee