mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-13 09:46:00 +00:00
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:
10
python/mscclpp/__init__.py
Normal file
10
python/mscclpp/__init__.py
Normal 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")
|
||||
Reference in New Issue
Block a user