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

[AutoTVM] Introduce PSOTuner for AutoTVM #16917

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

Junyi-99
Copy link

This commit introduces the Particle Swarm Optimization (PSO) tuner PSOTuner to AutoTVM, enhancing tuning efficiency by better exploring the configuration space compared to GATuner. Key additions include:

Major Changes:

  • PSOTuner Implementation: Implements the PSOTuner class using the PSO algorithm to manage particle initialization, position, velocity updates, and fitness evaluation for TVM tasks.
  • Integration into AutoTVM: The PSOTuner is now integrated into the AutoTVM tuner module, accessible like existing tuners such as GATuner.
  • Test Cases: Adds test cases to validate PSOTuner functionality and its effectiveness in various tuning scenarios.

Benefits:

  • Enhanced Tuning Efficiency: By leveraging PSO, which is known for its ability to find global optima in complex search spaces, the tuner is expected to provide better performance tuning outcomes.

This is an initial implementation, and feedback on its performance and functionality is highly encouraged to further refine this tuner.

This commit introduce a new PSOTuner class utilizing Particle Swarm Optimization for the AutoTVM tuning module.
@cbalint13
Copy link
Contributor

Hi @Junyi-99 ,

Thank you for the contribution !

Are there some numbers, advantages or a description on how this algorithm performs ?
If work is based on a paper or similar please make a reference in the code, let the users know more.

I add here few reference to code owners:
Cc @comaniac , @merrymercy , @vinx13 , @jcf94

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants