Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
robotsp opened this issue Aug 8, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@robotsp
Copy link

robotsp commented Aug 8, 2024

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 robotsp added the enhancement New feature or request label Aug 8, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant