diff --git a/benchmarks/gbench/mhp/CMakeLists.txt b/benchmarks/gbench/mhp/CMakeLists.txt
index 3a65166b6f..ccc68f5184 100644
--- a/benchmarks/gbench/mhp/CMakeLists.txt
+++ b/benchmarks/gbench/mhp/CMakeLists.txt
@@ -35,7 +35,7 @@ target_link_libraries(mhp-bench benchmark::benchmark cxxopts DR::mpi)
 
 # cmake-format: off
 add_executable(mhp-quick-bench mhp-bench.cpp
-  ../common/dot_product.cpp
+  stencil_2d.cpp
   )
 # cmake-format: on
 target_compile_definitions(mhp-quick-bench PRIVATE BENCH_MHP)
diff --git a/benchmarks/gbench/mhp/wave_equation.cpp b/benchmarks/gbench/mhp/wave_equation.cpp
index 4bfd968622..f4b6514585 100644
--- a/benchmarks/gbench/mhp/wave_equation.cpp
+++ b/benchmarks/gbench/mhp/wave_equation.cpp
@@ -657,6 +657,7 @@ int main(int argc, char *argv[]) {
     ("t,benchmark-mode", "Run a fixed number of time steps.", cxxopts::value<bool>()->default_value("false"))
     ("sycl", "Execute on SYCL device")
     ("f,fused-kernel", "Use fused kernels.", cxxopts::value<bool>()->default_value("false"))
+    ("device-memory", "Use device memory")
     ("h,help", "Print help");
   // clang-format on
 
@@ -673,7 +674,8 @@ int main(int argc, char *argv[]) {
     sycl::queue q = dr::mhp::select_queue();
     std::cout << "Run on: "
               << q.get_device().get_info<sycl::info::device::name>() << "\n";
-    dr::mhp::init(q);
+    dr::mhp::init(q, options.count("device-memory") ? sycl::usm::alloc::device
+                                                    : sycl::usm::alloc::shared);
 #else
     std::cout << "Sycl support requires icpx\n";
     exit(1);