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
2025-07-07 14:04:16 -07:00
committed by GitHub
parent ebd92a7b3c
commit 105382710a
2 changed files with 19 additions and 0 deletions
+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.