561 lines
11 KiB
C++
561 lines
11 KiB
C++
/*
|
|
Copyright (c) 2014 Intel Corporation. All Rights Reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
* Neither the name of Intel Corporation nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
|
|
#include <omp.h>
|
|
#include "offload.h"
|
|
#include "compiler_if_target.h"
|
|
|
|
// OpenMP API
|
|
|
|
void omp_set_default_device(int num) __GOMP_NOTHROW
|
|
{
|
|
}
|
|
|
|
int omp_get_default_device(void) __GOMP_NOTHROW
|
|
{
|
|
return mic_index;
|
|
}
|
|
|
|
int omp_get_num_devices() __GOMP_NOTHROW
|
|
{
|
|
return mic_engines_total;
|
|
}
|
|
|
|
// OpenMP API wrappers
|
|
|
|
static void omp_send_int_to_host(
|
|
void *ofld_,
|
|
int setting
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_out;
|
|
vars[0].ptr = &setting;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
static int omp_get_int_from_host(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
int setting;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_in;
|
|
vars[0].ptr = &setting;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
|
|
return setting;
|
|
}
|
|
|
|
void omp_set_num_threads_lrb(
|
|
void *ofld
|
|
)
|
|
{
|
|
int num_threads;
|
|
|
|
num_threads = omp_get_int_from_host(ofld);
|
|
omp_set_num_threads(num_threads);
|
|
}
|
|
|
|
void omp_get_max_threads_lrb(
|
|
void *ofld
|
|
)
|
|
{
|
|
int num_threads;
|
|
|
|
num_threads = omp_get_max_threads();
|
|
omp_send_int_to_host(ofld, num_threads);
|
|
}
|
|
|
|
void omp_get_num_procs_lrb(
|
|
void *ofld
|
|
)
|
|
{
|
|
int num_procs;
|
|
|
|
num_procs = omp_get_num_procs();
|
|
omp_send_int_to_host(ofld, num_procs);
|
|
}
|
|
|
|
void omp_set_dynamic_lrb(
|
|
void *ofld
|
|
)
|
|
{
|
|
int dynamic;
|
|
|
|
dynamic = omp_get_int_from_host(ofld);
|
|
omp_set_dynamic(dynamic);
|
|
}
|
|
|
|
void omp_get_dynamic_lrb(
|
|
void *ofld
|
|
)
|
|
{
|
|
int dynamic;
|
|
|
|
dynamic = omp_get_dynamic();
|
|
omp_send_int_to_host(ofld, dynamic);
|
|
}
|
|
|
|
void omp_set_nested_lrb(
|
|
void *ofld
|
|
)
|
|
{
|
|
int nested;
|
|
|
|
nested = omp_get_int_from_host(ofld);
|
|
omp_set_nested(nested);
|
|
}
|
|
|
|
void omp_get_nested_lrb(
|
|
void *ofld
|
|
)
|
|
{
|
|
int nested;
|
|
|
|
nested = omp_get_nested();
|
|
omp_send_int_to_host(ofld, nested);
|
|
}
|
|
|
|
void omp_set_schedule_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[2] = {0};
|
|
omp_sched_t kind;
|
|
int modifier;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_in;
|
|
vars[0].ptr = &kind;
|
|
|
|
vars[1].type.src = c_data;
|
|
vars[1].type.dst = c_data;
|
|
vars[1].direction.bits = c_parameter_in;
|
|
vars[1].ptr = &modifier;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 2, vars, NULL);
|
|
omp_set_schedule(kind, modifier);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_get_schedule_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[2] = {0};
|
|
omp_sched_t kind;
|
|
int modifier;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_out;
|
|
vars[0].ptr = &kind;
|
|
|
|
vars[1].type.src = c_data;
|
|
vars[1].type.dst = c_data;
|
|
vars[1].direction.bits = c_parameter_out;
|
|
vars[1].ptr = &modifier;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 2, vars, NULL);
|
|
omp_get_schedule(&kind, &modifier);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
// lock API functions
|
|
|
|
void omp_init_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_out;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_init_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_destroy_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_in;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_destroy_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_set_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_inout;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_set_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_unset_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_inout;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_unset_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_test_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[2] = {0};
|
|
omp_lock_target_t lock;
|
|
int result;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_inout;
|
|
vars[0].ptr = &lock;
|
|
|
|
vars[1].type.src = c_data;
|
|
vars[1].type.dst = c_data;
|
|
vars[1].direction.bits = c_parameter_out;
|
|
vars[1].ptr = &result;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 2, vars, NULL);
|
|
result = omp_test_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
// nested lock API functions
|
|
|
|
void omp_init_nest_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_nest_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_out;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_init_nest_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_destroy_nest_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_nest_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_in;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_destroy_nest_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_set_nest_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_nest_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_inout;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_set_nest_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_unset_nest_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[1] = {0};
|
|
omp_nest_lock_target_t lock;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_inout;
|
|
vars[0].ptr = &lock;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 1, vars, NULL);
|
|
omp_unset_nest_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
void omp_test_nest_lock_lrb(
|
|
void *ofld_
|
|
)
|
|
{
|
|
OFFLOAD ofld = (OFFLOAD) ofld_;
|
|
VarDesc vars[2] = {0};
|
|
omp_nest_lock_target_t lock;
|
|
int result;
|
|
|
|
vars[0].type.src = c_data;
|
|
vars[0].type.dst = c_data;
|
|
vars[0].direction.bits = c_parameter_inout;
|
|
vars[0].ptr = &lock;
|
|
|
|
vars[1].type.src = c_data;
|
|
vars[1].type.dst = c_data;
|
|
vars[1].direction.bits = c_parameter_out;
|
|
vars[1].ptr = &result;
|
|
|
|
OFFLOAD_TARGET_ENTER(ofld, 2, vars, NULL);
|
|
result = omp_test_nest_lock(&lock.lock);
|
|
OFFLOAD_TARGET_LEAVE(ofld);
|
|
}
|
|
|
|
// Target-side stubs for the host functions (to avoid unresolveds)
|
|
// These are needed for the offloadm table
|
|
|
|
void omp_set_num_threads_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
int num_threads
|
|
)
|
|
{
|
|
}
|
|
|
|
int omp_get_max_threads_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number
|
|
)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int omp_get_num_procs_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number
|
|
)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
void omp_set_dynamic_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
int num_threads
|
|
)
|
|
{
|
|
}
|
|
|
|
int omp_get_dynamic_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number
|
|
)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
void omp_set_nested_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
int num_threads
|
|
)
|
|
{
|
|
}
|
|
|
|
int omp_get_nested_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number
|
|
)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
void omp_set_schedule_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_sched_t kind,
|
|
int modifier
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_get_schedule_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_sched_t *kind,
|
|
int *modifier
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_init_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_destroy_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_set_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_unset_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
int omp_test_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_lock_target_t *lock
|
|
)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
void omp_init_nest_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_nest_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_destroy_nest_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_nest_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_set_nest_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_nest_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
void omp_unset_nest_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_nest_lock_target_t *lock
|
|
)
|
|
{
|
|
}
|
|
|
|
int omp_test_nest_lock_target(
|
|
TARGET_TYPE target_type,
|
|
int target_number,
|
|
omp_nest_lock_target_t *lock
|
|
)
|
|
{
|
|
return 0;
|
|
}
|