1 files changed, 61 insertions, 0 deletions
diff --git a/drivers/gpu/drm/i915/gt/intel_engine_stats.h b/drivers/gpu/drm/i915/gt/intel_engine_stats.h
new file mode 100644
index 000000000..8e762d683
--- /dev/null
+++ b/drivers/gpu/drm/i915/gt/intel_engine_stats.h
@@ -0,0 +1,61 @@
+/* SPDX-License-Identifier: MIT */
+/*
+ * Copyright © 2020 Intel Corporation
+ */
+
+#ifndef __INTEL_ENGINE_STATS_H__
+#define __INTEL_ENGINE_STATS_H__
+
+#include <linux/atomic.h>
+#include <linux/ktime.h>
+#include <linux/seqlock.h>
+
+#include "i915_gem.h" /* GEM_BUG_ON */
+#include "intel_engine.h"
+
+static inline void intel_engine_context_in(struct intel_engine_cs *engine)
+{
+	struct intel_engine_execlists_stats *stats = &engine->stats.execlists;
+	unsigned long flags;
+
+	if (stats->active) {
+		stats->active++;
+		return;
+	}
+
+	/* The writer is serialised; but the pmu reader may be from hardirq */
+	local_irq_save(flags);
+	write_seqcount_begin(&stats->lock);
+
+	stats->start = ktime_get();
+	stats->active++;
+
+	write_seqcount_end(&stats->lock);
+	local_irq_restore(flags);
+
+	GEM_BUG_ON(!stats->active);
+}
+
+static inline void intel_engine_context_out(struct intel_engine_cs *engine)
+{
+	struct intel_engine_execlists_stats *stats = &engine->stats.execlists;
+	unsigned long flags;
+
+	GEM_BUG_ON(!stats->active);
+	if (stats->active > 1) {
+		stats->active--;
+		return;
+	}
+
+	local_irq_save(flags);
+	write_seqcount_begin(&stats->lock);
+
+	stats->active--;
+	stats->total = ktime_add(stats->total,
+				 ktime_sub(ktime_get(), stats->start));
+
+	write_seqcount_end(&stats->lock);
+	local_irq_restore(flags);
+}
+
+#endif /* __INTEL_ENGINE_STATS_H__ */