How to add NVSHMEM?

[abcdabcd987]

Hi, I wonder if it's possible to add NVSHMEM (either to rules_cuda or just on my project for now)?

I saw many cuda libraries in the templates folder. I think those cc_import make sense to me. But there are still a few things that I don't know how to do:

1. Is there a cc_import equivalent for cuda_import? libnvshmem_device.a is a library that should be linked to the device code instead of host code.
2. How to export header files as well?

Here's the directory structure of NVSHMEM for reference:

nvshmem-dir.txt

Thanks!

[cloudhan]

See #244,

As for

a library that should be linked to the device code

is not supported at the moment. It is a known issue for me. The cufft_static suffers the same problem when device callback is needed.