From b485ddee3579ef99c2db7cb6c79263ffb89a061b Mon Sep 17 00:00:00 2001
From: Andrzej Janik <vosen@vosen.pl>
Date: Tue, 26 Mar 2024 20:20:01 +0100
Subject: [PATCH] Add failing test

---
 zluda/tests/primary_context.rs | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 zluda/tests/primary_context.rs

diff --git a/zluda/tests/primary_context.rs b/zluda/tests/primary_context.rs
new file mode 100644
index 0000000..39a68dc
--- /dev/null
+++ b/zluda/tests/primary_context.rs
@@ -0,0 +1,21 @@
+use crate::common::CudaDriverFns;
+use cuda_types::*;
+
+mod common;
+
+cuda_driver_test!(primary_context);
+
+unsafe fn primary_context<T: CudaDriverFns>(cuda: T) {
+    assert_eq!(cuda.cuInit(0), CUresult::CUDA_SUCCESS);
+    let mut flags = 0;
+    let mut active = 0;
+    assert_eq!(
+        cuda.cuDevicePrimaryCtxGetState(CUdevice_v1(0), &mut flags, &mut active),
+        CUresult::CUDA_SUCCESS
+    );
+    assert_eq!((0, 0), (flags, active));
+    assert_eq!(
+        cuda.cuDevicePrimaryCtxSetFlags_v2(CUdevice_v1(0), 1),
+        CUresult::CUDA_SUCCESS
+    );
+}