mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 04:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove outdated documentation section,

Browse Source 
CUPTI profiling now does support concurrent launches by default.

PiperOrigin-RevId: 617011144
This commit is contained in:
George Karpenkov 2024-03-18 18:24:18 -07:00 committed by jax authors
parent cfb968d266
commit 0e1b3e5ba6
1 changed files with 0 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
docs/profiling.md
View File

@ -216,16 +216,6 @@ from a running program.
You can also use the `memory_viewer`, `op_profile`, and `graph_viewer`
tools.<br /><br />
### Concurrent kernel tracing on GPU
By default, traces are captured on GPU in a mode that prevents CUDA kernels from
running concurrently. This allows for more accurate kernel timings, but removes
any concurrency between streams (for example, between compute and
communication). To enable concurrent kernel tracing, set the environment
variable `TF_GPU_CUPTI_FORCE_CONCURRENT_KERNEL=1` when launching the JAX
program.
### Adding custom trace events
By default, the events in the trace viewer are mostly low-level internal JAX