chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #2271 from xela-95:patch-1

Browse Source 
PiperOrigin-RevId: 703493322
Change-Id: I5034ecdbc2e129986bb1e61930637fd2c24b96a6
This commit is contained in:
Copybara-Service
2024-12-06 07:26:47 -08:00
parent fa8af0eea9 342a72839a
commit 15cb1fdfb4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/mjx.rst
+1 -1
View File
@@ -416,4 +416,4 @@ The following environment variables should be set:
``XLA_FLAGS=--xla_gpu_triton_gemm_any=true`` ``XLA_FLAGS=--xla_gpu_triton_gemm_any=true``
This enables the Triton-based GEMM (matmul) emitter for any GEMM that it supports. This can yield a 30% speedup on This enables the Triton-based GEMM (matmul) emitter for any GEMM that it supports. This can yield a 30% speedup on
NVIDIA GPUs. If you have multiple GPUs, you may also benefit from enabling flags related to NVIDIA GPUs. If you have multiple GPUs, you may also benefit from enabling flags related to
`communciation between GPUs <https://jax.readthedocs.io/en/latest/gpu_performance_tips.html>`__. `communication between GPUs <https://jax.readthedocs.io/en/latest/gpu_performance_tips.html>`__.