tvm_tirx.support.nvcc.find_nvshmem_paths

tvm_tirx.support.nvcc.find_nvshmem_paths#

tvm_tirx.support.nvcc.find_nvshmem_paths()#

Searches for the NVSHMEM include and library directories.

Returns:

A tuple containing the path to the include directory and the library directory.

Return type:

tuple[str, str]