ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[BUG]: The placement policies between Bert example and the stable ColAI are not compatible

Open LotsoTeddy opened this issue 2 years ago • 1 comments

🐛 Describe the bug

The placement policy of Gemini plugin in Bert example is "cuda", which is as follows:

image

However, this policy is not supported by the ColAI (main branch). Existing policy includes "auto" and "static".

image

Therefore, the error is:

image

Correct the policy from "cuda" to "auto" can solve this problem and successfully testing:

image

Thanks.

Environment

No response

LotsoTeddy avatar Dec 23 '23 05:12 LotsoTeddy

Thank you, we will fix this issue as soon as possible.

flybird11111 avatar Dec 23 '23 08:12 flybird11111