fix

Author: wsmoses

deps/ReactantExtra/WORKSPACE

@@ -52,9 +52,17 @@ hedron_compile_commands_setup_transitive_transitive_transitive()
 
 load("@enzyme_ad//:workspace.bzl", "ENZYME_COMMIT", "ENZYME_SHA256", "JAX_COMMIT", "JAX_SHA256", "XLA_PATCHES")
 
-XLA_PATCHES = XLA_PATCHES + [
+CUPTI_OLD = [
 """
-sed -i.bak0 "/CUPTI_DRIVER_TRACE_CBID_cuGraphAddNode/d" "/CUPTI_DRIVER_TRACE_CBID_cuGraphAddNode_v2/d" "s/cupti_driver_cbid/cupti/g" xla/backends/profiler/gpu/cupti_tracer.cc
+sed -i.bak0 "s/case CUPTI_DRIVER_TRACE_CBID_cuGraphAddNode:/d" "/CUPTI_DRIVER_TRACE_CBID_cuGraphAddNode_v2/d" "s/cupti_driver_cbid/cupti/g" xla/backends/profiler/gpu/cupti_tracer.cc
+"""
+]
+
+CUPTI_NEW = []
+
+XLA_PATCHES = XLA_PATCHES + CUPTI_NEW + [
+"""
+sed -i.bak0 "s/cupti_driver_cbid/cupti/g" xla/backends/profiler/gpu/cupti_tracer.cc
 """,
     """
 sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.bzl -exec sed -i.bak0 's\\/HAVE_LINK_H=1\\/HAVE_LINK_H=0\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl