You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/gpus.rst
+12Lines changed: 12 additions & 0 deletions
Original file line number
Diff line number
Diff line change
@@ -20,6 +20,18 @@ The HAPI Charm4py API is:
20
20
For now, ``charm.hapiAddCudaCallback`` only supports numba and torch streams as input. This function inserts a callback
21
21
into the stream such that when the callback is reached, the corresponding Charm4py future is set.
22
22
23
+
Enabling HAPI
24
+
--------
25
+
To build Charm4py with HAPI support, add "cuda" to the Charm build options and follow the steps to build Charm4py from source:
26
+
27
+
.. code-block:: shell
28
+
29
+
export CHARM_EXTRA_BUILD_OPTS="cuda"
30
+
pip install .
31
+
32
+
.. warning::
33
+
34
+
To ensure that the underlying Charm build has Cuda enabled, remove any pre-existing builds in charm_src/charm before setting the Cuda option and running install.
0 commit comments