Refactor setup.py to use get_pybind_include

[sakgoyal]

What does this PR do ?

Try to fix bugs when installing on windows with uv. see #2609

Pre-checks

• I want this PR in a versioned release and have added the appropriate Milestone (e.g., Core 0.8)
• I have added relevant unit tests
• I have added relevant functional tests
• I have added proper typing to my code Typing guidelines
• I have added relevant documentation
• I have run the autoformatter.sh on my PR

[copy-pr-bot]

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

[sakgoyal]

@ko3n1g Sorry for the ping. Just wanted to know when I should expect this to get merged

[sakgoyal]

@Skylion007 do you know how I can get ko3n1g to review this?

[ko3n1g]

/ok to test 935a5bb

[ko3n1g]

Hey @sakgoyal , thanks for your contribution! It looks like CI is failing, I will try to take a look. However as vacations are upcoming it might get delayed a bit.

[sakgoyal]

These failures don't seem to be from my PR. seems that something else went wrong that caused this to fail.

Also, should I update this PR (rebase / merge to HEAD) or leave it as it is? Not sure what the policy is for PRs

[ko3n1g]

These workflows are passing in the target branch: https://github.com/NVIDIA/Megatron-LM/actions/workflows/build-test-publish-wheel.yml?query=branch%3Adev

So we need do need to take a closer look..

[ko3n1g]

/ok to test e9641e9

[github-actions]

Thank you for your contribution!

NVIDIA Megatron-LM is currently transitioning to development on Github. We will aim to review your PR after we complete our transition and stabilize our Github development process.

Thank you for your understanding.

[sakgoyal]

should I do an update with rebase through github for this PR? or should I leave it alone? I'd like to run CI again on this if possible.

Im asking because I don't understand some of the CI errors about the whl not being valid as my changes should not be affecting that.