Add Python get_include() (#141)

Introduces a mscclpp.get_include() in the Python module.
The extension module is now named _mscclpp so that we can have
Python code in the mscclpp module.
Also does some miscellaneous cleanup.
This commit is contained in:
Olli Saarikivi
2023-07-25 10:23:16 -07:00
committed by GitHub
parent 9a488f0da2
commit 4865b2017b
6 changed files with 26 additions and 14 deletions

View File

@@ -0,0 +1,10 @@
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from ._mscclpp import *
import os as _os
def get_include():
"""Return the directory that contains the MSCCL++ headers."""
return _os.path.join(_os.path.dirname(__file__), "include")