DeepSpeed

bc7778ea - Fix the workspace allocation for the transformer kernel (#1397)

Fix the workspace allocation for the transformer kernel (#1397) * fix the workspace allocation for the transformer kernel * change layer-id type & rm one unit test due to OOM