Skip to content

Commit 4ea5805

Browse files
jingxu10pytorchmergebot
authored andcommitted
update aotinductor doc for XPU support (pytorch#149299)
as title. Since the AOTInductor feature starting from 2.7 works on Intel GPU, add the related contents into its doc. Pull Request resolved: pytorch#149299 Approved by: https://github.com/guangyey, https://github.com/desertfire
1 parent ccd5d81 commit 4ea5805

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

docs/source/torch.compiler_aot_inductor.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,8 @@ package.
3838
the following code will compile the model into a shared library for CUDA execution.
3939
Otherwise, the compiled artifact will run on CPU. For better performance during CPU inference,
4040
it is suggested to enable freezing by setting ``export TORCHINDUCTOR_FREEZING=1``
41-
before running the Python script below.
41+
before running the Python script below. The same behavior works in an environment with Intel®
42+
GPU as well.
4243

4344
.. code-block:: python
4445

0 commit comments

Comments
 (0)