73 lines
2.4 KiB
Diff
73 lines
2.4 KiB
Diff
From bb52aeacc6dfab2355249b7b5beb72c2761ec319 Mon Sep 17 00:00:00 2001
|
|
From: Tom Rix <trix@redhat.com>
|
|
Date: Fri, 29 Sep 2023 06:25:23 -0700
|
|
Subject: [PATCH 3/6] Stub in kineto ActivityType
|
|
|
|
There is an error with kineto is not used, the shim still
|
|
requires the ActivityTYpe.h header to get the enum Activity type.
|
|
So cut-n-paste just enough of the header in to do this.
|
|
|
|
Signed-off-by: Tom Rix <trix@redhat.com>
|
|
---
|
|
torch/csrc/profiler/kineto_shim.h | 44 +++++++++++++++++++++++++++++++
|
|
1 file changed, 44 insertions(+)
|
|
|
|
diff --git a/torch/csrc/profiler/kineto_shim.h b/torch/csrc/profiler/kineto_shim.h
|
|
index 2a410719a1..7d6525befd 100644
|
|
--- a/torch/csrc/profiler/kineto_shim.h
|
|
+++ b/torch/csrc/profiler/kineto_shim.h
|
|
@@ -12,7 +12,51 @@
|
|
#undef USE_KINETO
|
|
#endif
|
|
|
|
+#ifdef USE_KINETO
|
|
#include <ActivityType.h>
|
|
+#else
|
|
+namespace libkineto {
|
|
+// copied from header
|
|
+/*
|
|
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
+ * All rights reserved.
|
|
+ *
|
|
+ * This source code is licensed under the BSD-style license found in the
|
|
+ * LICENSE file in the root directory of this source tree.
|
|
+ */
|
|
+
|
|
+// Note : All activity types are not enabled by default. Please add them
|
|
+// at correct position in the enum
|
|
+enum class ActivityType {
|
|
+ // Activity types enabled by default
|
|
+ CPU_OP = 0, // cpu side ops
|
|
+ USER_ANNOTATION,
|
|
+ GPU_USER_ANNOTATION,
|
|
+ GPU_MEMCPY,
|
|
+ GPU_MEMSET,
|
|
+ CONCURRENT_KERNEL, // on-device kernels
|
|
+ EXTERNAL_CORRELATION,
|
|
+ CUDA_RUNTIME, // host side cuda runtime events
|
|
+ CUDA_DRIVER, // host side cuda driver events
|
|
+ CPU_INSTANT_EVENT, // host side point-like events
|
|
+ PYTHON_FUNCTION,
|
|
+ OVERHEAD, // CUPTI induced overhead events sampled from its overhead API.
|
|
+
|
|
+ // Optional Activity types
|
|
+ CUDA_SYNC, // synchronization events between runtime and kernels
|
|
+ GLOW_RUNTIME, // host side glow runtime events
|
|
+ MTIA_RUNTIME, // host side MTIA runtime events
|
|
+ CUDA_PROFILER_RANGE, // CUPTI Profiler range for performance metrics
|
|
+ MTIA_CCP_EVENTS, // MTIA ondevice CCP events
|
|
+ HPU_OP, // HPU host side runtime event
|
|
+ XPU_RUNTIME, // host side xpu runtime events
|
|
+
|
|
+ ENUM_COUNT, // This is to add buffer and not used for any profiling logic. Add your new type before it.
|
|
+ OPTIONAL_ACTIVITY_TYPE_START = CUDA_SYNC,
|
|
+};
|
|
+}
|
|
+
|
|
+#endif
|
|
|
|
#include <torch/csrc/Export.h>
|
|
#include <torch/csrc/profiler/api.h>
|
|
--
|
|
2.41.0
|
|
|