1 files changed, 15 insertions, 5 deletions

diff --git a/Makefile b/Makefile
index 6abc124..b91f6b9 100644
--- a/Makefile
+++ b/Makefile
@@ -14,14 +14,24 @@ libsmctrl.a: libsmctrl.c libsmctrl.h
         ar rcs $@ libsmctrl.o
 
 # Use static linking with tests to avoid LD_LIBRARY_PATH issues
-libsmctrl_test_gpc_info: libsmctrl_test_gpc_info.c libsmctrl.a
+libsmctrl_test_gpc_info: libsmctrl_test_gpc_info.c libsmctrl.a testbench.h
         $(CC) $< -o $@ -g -L. -l:libsmctrl.a $(LDFLAGS)
 
-libsmctrl_test_global_mask: libsmctrl_test_global_mask.cu libsmctrl.a
-        $(NVCC) $< -o $@ -g -L. -l:libsmctrl.a $(LDFLAGS)
+libsmctrl_test_mask_shared.o: libsmctrl_test_mask_shared.cu testbench.h
+        $(NVCC) $< -c -g
 
-tests: libsmctrl_test_gpc_info libsmctrl_test_global_mask
+libsmctrl_test_global_mask: libsmctrl_test_global_mask.c libsmctrl.a libsmctrl_test_mask_shared.o
+        $(NVCC) $@.c -o $@ libsmctrl_test_mask_shared.o -g -L. -l:libsmctrl.a $(LDFLAGS)
+
+libsmctrl_test_stream_mask: libsmctrl_test_stream_mask.c libsmctrl.a libsmctrl_test_mask_shared.o
+        $(NVCC) $@.c -o $@ libsmctrl_test_mask_shared.o -g -L. -l:libsmctrl.a $(LDFLAGS)
+
+libsmctrl_test_next_mask: libsmctrl_test_next_mask.c libsmctrl.a libsmctrl_test_mask_shared.o
+        $(NVCC) $@.c -o $@ libsmctrl_test_mask_shared.o -g -L. -l:libsmctrl.a $(LDFLAGS)
+
+tests: libsmctrl_test_gpc_info libsmctrl_test_global_mask libsmctrl_test_stream_mask libsmctrl_test_next_mask
 
 clean:
         rm -f libsmctrl.so libsmctrl.a libsmctrl_test_gpu_info \
-              libsmctrl_test_global_mask
+              libsmctrl_test_mask_shared.o libmsctrl_test_global_mask \
+              libsmctrl_test_stream_mask libmsctrl_test_next_mask