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``
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
`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>`__.