1 files changed, 9 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/barrier.h b/drivers/gpu/nvgpu/include/nvgpu/barrier.h
index 01bd3023..03a14a99 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/barrier.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/barrier.h
@@ -47,4 +47,13 @@
 
 #define NV_ACCESS_ONCE(x)       __NV_ACCESS_ONCE(x)
 
+/*
+ * Sometimes we want to prevent speculation.
+ */
+#ifdef __NVGPU_PREVENT_UNTRUSTED_SPECULATION
+#define nvgpu_speculation_barrier()     __nvgpu_speculation_barrier()
+#else
+#define nvgpu_speculation_barrier()
+#endif
+
 #endif /* __NVGPU_BARRIER_H__ */