diff --git a/catch/unit/device/hipDeviceReset.cc b/catch/unit/device/hipDeviceReset.cc index b65155db6f..6136a71a2b 100644 --- a/catch/unit/device/hipDeviceReset.cc +++ b/catch/unit/device/hipDeviceReset.cc @@ -69,7 +69,7 @@ TEST_CASE("Unit_hipDeviceReset_Positive_Basic") { : hipSharedMemBankSizeFourByte); REQUIRE((shared_mem_config_ret == hipSuccess || shared_mem_config_ret == hipErrorNotSupported)); - HIP_CHECK(hipSetDeviceFlags(flags_before ^ (1u << 2))); + HIP_CHECK(hipSetDeviceFlags(hipDeviceScheduleBlockingSync)); HIP_CHECK(hipDeviceReset()); @@ -134,7 +134,7 @@ TEST_CASE("Unit_hipDeviceReset_Positive_Threaded") { REQUIRE((shared_mem_config_ret == hipSuccess || shared_mem_config_ret == hipErrorNotSupported)); - HIP_CHECK(hipSetDeviceFlags(flags_before ^ (1u << 2))); + HIP_CHECK(hipSetDeviceFlags(hipDeviceScheduleBlockingSync)); std::thread([] { HIP_CHECK_THREAD(hipSetDevice(0));