// REQUIRES: aspect-usm_shared_allocations
// RUN: %{build} -o %t.out
// RUN: env SYCL_UR_TRACE=2 %{run} %t.out 2>&1 | FileCheck %s

// Tests the enqueue free function mem_advise.

#include <sycl/detail/core.hpp>
#include <sycl/ext/oneapi/experimental/enqueue_functions.hpp>
#include <sycl/usm.hpp>

namespace oneapiext = sycl::ext::oneapi::experimental;

constexpr size_t N = 1024;

int main() {
  sycl::context Context;
  sycl::queue Q(Context, sycl::default_selector_v);
  int *Memory = sycl::malloc_shared<int>(N, Q);

  constexpr size_t ChunkSize = N / 3;

  oneapiext::mem_advise(Q, Memory, ChunkSize, 0);

  oneapiext::submit(Q, [&](sycl::handler &CGH) {
    oneapiext::mem_advise(CGH, Memory + ChunkSize, ChunkSize, 0);
  });

  sycl::event E = oneapiext::submit_with_event(Q, [&](sycl::handler &CGH) {
    oneapiext::mem_advise(CGH, Memory + ChunkSize * 2, ChunkSize, 0);
  });

  E.wait();
  Q.wait();
  sycl::free(Memory, Q);

  return 0;
}

// CHECK-COUNT-3: <--- urEnqueueUSMAdvise
// CHECK-NOT: <--- urEnqueueUSMAdvise