Add host API to query Device side context detail (#183)

* API support for enabling rocshmem bitcode integration

* move implementation to along with host side APIs
This commit is contained in:
Dimple Prajapati authored and GitHub committed 2025-07-07 16:04:16 -05:00
1 parent ebd92a7b3c
commit 105382710a
2 files changed
+19

No files matched your search

+11
View File
@@ -65,6 +65,17 @@ constexpr char VERSION[] = "3.0.0";
*/
__host__ void rocshmem_init(MPI_Comm comm = MPI_COMM_WORLD);
/**
* @brief Query rocSHMEM context from host API
*
* @param[out] ctx Returns ROCSHMEM_CTX_DEFAULT device pointer that users
* can query from one instance of rocshmem host library and
* use use later for dynamic module initialization in
* kernel bitcode device library in the same application
*/
__host__ void * rocshmem_get_device_ctx();
/**
* @brief Initialize the rocSHMEM runtime and underlying transport layer
* with an attempt to enable the requested thread support.