ColossalAI
ColossalAI copied to clipboard
[FEATURE]: How to skip a custom node from generating strategies in colossal-auto?
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?