1 files changed, 126 insertions, 0 deletions
diff --git a/drivers/gator/gator_trace_gpu.c b/drivers/gator/gator_trace_gpu.c
new file mode 100644
index 00000000000..921932c1fb1
--- /dev/null
+++ b/drivers/gator/gator_trace_gpu.c
@@ -0,0 +1,126 @@
+/**
+ * Copyright (C) ARM Limited 2010-2012. All rights reserved.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ */
+
+#include "gator.h"
+
+#include <linux/slab.h>
+#include <linux/module.h>
+#include <linux/time.h>
+#include <linux/math64.h>
+
+#ifdef MALI_SUPPORT
+#include "linux/mali_linux_trace.h"
+#endif
+#include "gator_trace_gpu.h"
+
+#define ACTIVITY_START  1
+#define ACTIVITY_STOP   2
+
+/* Note whether tracepoints have been registered */
+static int mali_trace_registered;
+static int gpu_trace_registered;
+
+#define GPU_START			1
+#define GPU_STOP			2
+
+#define GPU_UNIT_VP			1
+#define GPU_UNIT_FP			2
+#define GPU_UNIT_CL			3
+
+#ifdef MALI_SUPPORT
+
+enum components {
+    COMPONENT_VP0 = 1,
+    COMPONENT_FP0 = 5,
+    COMPONENT_FP1,
+    COMPONENT_FP2,
+    COMPONENT_FP3,
+    COMPONENT_FP4,
+    COMPONENT_FP5,
+    COMPONENT_FP6,
+    COMPONENT_FP7,
+};
+
+GATOR_DEFINE_PROBE(mali_timeline_event, TP_PROTO(unsigned int event_id, unsigned int d0, unsigned int d1, unsigned int d2, unsigned int d3, unsigned int d4))
+{
+	unsigned int component, state;
+	int tgid = 0, pid = 0;
+
+	// do as much work as possible before disabling interrupts
+	component = (event_id >> 16) & 0xFF; // component is an 8-bit field
+	state = (event_id >> 24) & 0xF;      // state is a 4-bit field
+
+	if ((component == COMPONENT_VP0) || (component >= COMPONENT_FP0 && component <= COMPONENT_FP7)) {
+		if (state == ACTIVITY_START || state == ACTIVITY_STOP) {
+			unsigned int type = (state == ACTIVITY_START) ? GPU_START : GPU_STOP;
+			unsigned int unit = (component < COMPONENT_FP0) ? GPU_UNIT_VP : GPU_UNIT_FP;
+			unsigned int core = (component < COMPONENT_FP0) ? component - COMPONENT_VP0 : component - COMPONENT_FP0;
+			if (state == ACTIVITY_START) {
+				tgid = d0;
+				pid = d1;
+			}
+
+			marshal_sched_gpu(type, unit, core, tgid, pid);
+    	}
+    }
+}
+#endif
+
+GATOR_DEFINE_PROBE(gpu_activity_start, TP_PROTO(int gpu_unit, int gpu_core, struct task_struct *p))
+{
+	marshal_sched_gpu(GPU_START, gpu_unit, gpu_core, (int)p->tgid, (int)p->pid);
+}
+
+GATOR_DEFINE_PROBE(gpu_activity_stop, TP_PROTO(int gpu_unit, int gpu_core))
+{
+	marshal_sched_gpu(GPU_STOP, gpu_unit, gpu_core, 0, 0);
+}
+
+int gator_trace_gpu_start(void)
+{
+	/*
+	 * Returns nonzero for installation failed
+	 * Absence of gpu trace points is not an error
+	 */
+
+	gpu_trace_registered = mali_trace_registered = 0;
+
+#ifdef MALI_SUPPORT
+    if (!GATOR_REGISTER_TRACE(mali_timeline_event)) {
+    	mali_trace_registered = 1;
+    }
+#endif
+
+    if (!mali_trace_registered) {
+        if (GATOR_REGISTER_TRACE(gpu_activity_start)) {
+        	return 0;
+        }
+        if (GATOR_REGISTER_TRACE(gpu_activity_stop)) {
+        	GATOR_UNREGISTER_TRACE(gpu_activity_start);
+        	return 0;
+        }
+        gpu_trace_registered = 1;
+    }
+
+	return 0;
+}
+
+void gator_trace_gpu_stop(void)
+{
+#ifdef MALI_SUPPORT
+	if (mali_trace_registered) {
+		GATOR_UNREGISTER_TRACE(mali_timeline_event);
+	}
+#endif
+	if (gpu_trace_registered) {
+		GATOR_UNREGISTER_TRACE(gpu_activity_stop);
+		GATOR_UNREGISTER_TRACE(gpu_activity_start);
+	}
+
+	gpu_trace_registered = mali_trace_registered = 0;
+}