1 files changed, 9 insertions, 0 deletions

diff --git a/libsmctrl_test_mask_shared.cu b/libsmctrl_test_mask_shared.cu
index c3817fa..f3e8933 100644
--- a/libsmctrl_test_mask_shared.cu
+++ b/libsmctrl_test_mask_shared.cu
@@ -87,9 +87,18 @@ int test_constrained_size_and_location(enum partitioning_type part_type) {
       case PARTITION_STREAM:
         libsmctrl_set_stream_mask_ext(stream, mask);
         break;
+      case PARTITION_STREAM_OVERRIDE:
+        libsmctrl_set_global_mask(~mask);
+        libsmctrl_set_stream_mask_ext(stream, mask);
+        break;
       case PARTITION_NEXT:
         libsmctrl_set_next_mask(mask);
         break;
+      case PARTITION_NEXT_OVERRIDE:
+        libsmctrl_set_global_mask(~mask);
+        libsmctrl_set_stream_mask_ext(stream, ~mask);
+        libsmctrl_set_next_mask(mask);
+        break;
       default:
         error(1, 0, "Shared test core called with unrecognized partitioning type.");
     }