// RUN: %{build} -o %t1.out
// RUN: %{run-unfiltered-devices} %t1.out

// REQUIRES: cpu, gpu

//==----------------- multictxt.cpp - Multi Context USM test ---------------==//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include <sycl/detail/core.hpp>
#include <sycl/usm.hpp>

using namespace sycl;

// The multictxt test here is a sanity check that USM selects the right
// implementation when presented with multiple contexts. The extra context
// only needs to exist for this test to do its job.

void GpuCpuCpu() {
  queue gpu_q(gpu_selector_v);
  queue cpu_q(cpu_selector_v);
  device dev = cpu_q.get_device();
  context ctx = cpu_q.get_context();
  if (dev.get_info<info::device::usm_shared_allocations>()) {
    void *ptr = malloc_shared(128, dev, ctx);

    free(ptr, ctx);
  }
}

void CpuGpuGpu() {
  queue cpu_q(cpu_selector_v);
  queue gpu_q(gpu_selector_v);
  device dev = gpu_q.get_device();
  context ctx = gpu_q.get_context();

  if (dev.get_info<info::device::usm_shared_allocations>()) {
    void *ptr = malloc_shared(128, dev, ctx);

    free(ptr, ctx);
  }
}

void GpuCpuGpu() {
  queue gpu_q(gpu_selector_v);
  queue cpu_q(cpu_selector_v);
  device dev = gpu_q.get_device();
  context ctx = gpu_q.get_context();

  if (dev.get_info<info::device::usm_shared_allocations>()) {
    void *ptr = malloc_shared(128, dev, ctx);

    free(ptr, ctx);
  }
}

int main() {
  GpuCpuCpu();
  CpuGpuGpu();
  GpuCpuGpu();

  return 0;
}