Skip to content

Commit aa47a21

Browse files
committed
update
1 parent 5c9d78b commit aa47a21

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

advanced_source/python_extension_autoload.rst

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,8 @@ Now the ``torch_foo`` module can be imported when running import torch:
6161
Examples
6262
^^^^^^^^
6363

64+
TODO: take HPU and NPU as examples
65+
6466
`habana_frameworks.torch`_ is a Python package that enables users to run
6567
PyTorch programs on Intel Gaudi via the PyTorch ``HPU`` device key.
6668
``import habana_frameworks.torch`` is no longer necessary after this mechanism
@@ -96,7 +98,7 @@ as ``npu`` to the end users.
9698
How it works
9799
------------
98100

99-
.. image:: ../_static/img/python_backend_autoload_impl.png
101+
.. image:: ../_static/img/python_extension_autoload_impl.png
100102
:alt: Autoloading implementation
101103
:align: center
102104

0 commit comments

Comments
 (0)