mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 01:10:22 +00:00
4865b2017b
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.
11 lines
266 B
Python
11 lines
266 B
Python
# 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")
|