ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[FEATURE]: How to skip a custom node from generating strategies in colossal-auto?

Open robotsp opened this issue 1 year ago • 0 comments

Describe the feature

I build a custom network with some custom operators, e.g., flash-attention. But I found neither colotracer nor node_handler can deal with them successfully. But I don't want to write strategies for them actually. How can I skip them or I must need to add some codes to adapt?

robotsp avatar Aug 08 '24 12:08 robotsp