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

fix member access to packed CUDA struct

Browse Source
This commit is contained in:
H. Vetinari 2025-02-22 16:39:41 +11:00
parent 908ff49e22
commit 91cae595e4
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
jaxlib/mosaic/gpu/mosaic_gpu_ext.cc
View File

@ -182,8 +182,9 @@ void callback_complete(CUcontext context, uint32_t streamId,
// Convert integer nanoseconds to floating point milliseconds to match
// the interface of the events-based profiler.
double duration_ms = (kernel->end - kernel->start) / 1e6;
const char* kernel_name = kernel->name;
profiler_state.timings.push_back(
std::make_tuple(kernel->name, duration_ms));
std::make_tuple(kernel_name, duration_ms));
}
} else if (status == CUPTI_ERROR_MAX_LIMIT_REACHED) {
// no more records available