Fix #2343.

JuliaGPU · Apr 25, 2024 · 6659669 · 6659669
1 parent 5dd6bb2
commit 6659669
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/compiler/reflection.jl b/src/compiler/reflection.jl
@@ -88,6 +88,16 @@ function code_sass(f::Base.Callable, io::IO=stdout; raw::Bool=false)
         return
     end
 
+    # NVIDIA bug #4604961: CUPTI in CUDA 12.4 Update 1 does not capture profiled events
+    #                      unless the activity API is first activated
+    if runtime_version() == v"12.4"
+        cfg = CUPTI.ActivityConfig([CUPTI.CUPTI_ACTIVITY_KIND_CONCURRENT_KERNEL,
+                                    CUPTI.CUPTI_ACTIVITY_KIND_INTERNAL_LAUNCH_API])
+        CUPTI.enable!(cfg) do
+            # do nothing
+        end
+    end
+
     seen_modules = Set{UInt32}()
     cfg = CUPTI.CallbackConfig([CUPTI.CUPTI_CB_DOMAIN_RESOURCE]) do domain, id, data
         # only process relevant callbacks