truffle: src/gpu/ptx/gpu_ptx.cpp annotate

annotate src/gpu/ptx/gpu_ptx.cpp @ 10806:f272fe8a2f9a

move LockElimination to MidTier

author	Lukas Stadler <lukas.stadler@jku.at>
date	Thu, 18 Jul 2013 13:16:37 +0200
parents	b0b368d38b40
children

rev	line source
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	1 /*
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	2 * Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved.
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	4 *
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	5 * This code is free software; you can redistribute it and/or modify it
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	6 * under the terms of the GNU General Public License version 2 only, as
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	7 * published by the Free Software Foundation.
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	8 *
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	9 * This code is distributed in the hope that it will be useful, but WITHOUT
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	12 * version 2 for more details (a copy is included in the LICENSE file that
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	13 * accompanied this code).
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	14 *
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	15 * You should have received a copy of the GNU General Public License version
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	16 * 2 along with this work; if not, write to the Free Software Foundation,
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	18 *
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	20 * or visit www.oracle.com if you need additional information or have any
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	21 * questions.
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	22 *
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	23 */
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	24
9485 62838eadbf56 Windows build fix for PTX code Andreas Woess <andreas.woess@jku.at> parents: 9430 diff changeset	25 #include "precompiled.hpp"
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	26 #include "runtime/gpu.hpp"
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	27 #include "utilities/globalDefinitions.hpp"
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	28 #include "utilities/ostream.hpp"
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	29
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	30 void * gpu::Ptx::_device_context;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	31
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	32 gpu::Ptx::cuda_cu_init_func_t gpu::Ptx::_cuda_cu_init;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	33 gpu::Ptx::cuda_cu_ctx_create_func_t gpu::Ptx::_cuda_cu_ctx_create;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	34 gpu::Ptx::cuda_cu_ctx_detach_func_t gpu::Ptx::_cuda_cu_ctx_detach;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	35 gpu::Ptx::cuda_cu_ctx_synchronize_func_t gpu::Ptx::_cuda_cu_ctx_synchronize;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	36 gpu::Ptx::cuda_cu_device_get_count_func_t gpu::Ptx::_cuda_cu_device_get_count;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	37 gpu::Ptx::cuda_cu_device_get_name_func_t gpu::Ptx::_cuda_cu_device_get_name;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	38 gpu::Ptx::cuda_cu_device_get_func_t gpu::Ptx::_cuda_cu_device_get;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	39 gpu::Ptx::cuda_cu_device_compute_capability_func_t gpu::Ptx::_cuda_cu_device_compute_capability;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	40 gpu::Ptx::cuda_cu_launch_kernel_func_t gpu::Ptx::_cuda_cu_launch_kernel;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	41 gpu::Ptx::cuda_cu_module_get_function_func_t gpu::Ptx::_cuda_cu_module_get_function;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	42 gpu::Ptx::cuda_cu_module_load_data_ex_func_t gpu::Ptx::_cuda_cu_module_load_data_ex;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	43
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	44 void gpu::probe_linkage() {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	45 #ifdef __APPLE__
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	46 set_gpu_linkage(gpu::Ptx::probe_linkage_apple());
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	47 #else
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	48 set_gpu_linkage(false);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	49 #endif
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	50 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	51
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	52 void gpu::initialize_gpu() {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	53 if (gpu::has_gpu_linkage()) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	54 set_initialized(gpu::Ptx::initialize_gpu());
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	55 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	56 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	57
10566 a6632ef9c84d GPU generate_kernel return for Cuda function Morris Meyer <morris.meyer@oracle.com> parents: 10563 diff changeset	58 void * gpu::generate_kernel(unsigned char code, int code_len, const char name) {
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	59 if (gpu::has_gpu_linkage()) {
10566 a6632ef9c84d GPU generate_kernel return for Cuda function Morris Meyer <morris.meyer@oracle.com> parents: 10563 diff changeset	60 return (gpu::Ptx::generate_kernel(code, code_len, name));
a6632ef9c84d GPU generate_kernel return for Cuda function Morris Meyer <morris.meyer@oracle.com> parents: 10563 diff changeset	61 } else {
a6632ef9c84d GPU generate_kernel return for Cuda function Morris Meyer <morris.meyer@oracle.com> parents: 10563 diff changeset	62 return NULL;
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	63 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	64 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	65
10577 9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	66 bool gpu::execute_kernel(address kernel) {
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	67 if (gpu::has_gpu_linkage()) {
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	68 return (gpu::Ptx::execute_kernel(kernel));
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	69 } else {
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	70 return false;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	71 }
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	72 }
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	73
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	74 #define __CUDA_API_VERSION 5000
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	75
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	76 bool gpu::Ptx::initialize_gpu() {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	77 int status = _cuda_cu_init(0, __CUDA_API_VERSION);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	78 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	79 tty->print_cr("gpu_ptx::_cuda_cu_init: %d", status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	80 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	81
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	82 int device_count = 0;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	83 status = _cuda_cu_device_get_count(&device_count);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	84 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	85 tty->print_cr("gpu_ptx::_cuda_cu_device_get_count(%d): %d", device_count, status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	86 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	87
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	88 int device_id = 0, cu_device = 0;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	89 status = _cuda_cu_device_get(&cu_device, device_id);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	90 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	91 tty->print_cr("gpu_ptx::_cuda_cu_device_get(%d): %d", cu_device, status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	92 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	93
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	94 int major, minor;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	95 status = _cuda_cu_device_compute_capability(&major, &minor, cu_device);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	96 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	97 tty->print_cr("gpu_ptx::_cuda_cu_device_compute_capability(major %d, minor %d): %d",
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	98 major, minor, status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	99 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	100
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	101 char device_name[256];
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	102 status = _cuda_cu_device_get_name(device_name, 256, cu_device);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	103 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	104 tty->print_cr("gpu_ptx::_cuda_cu_device_get_name(%s): %d", device_name, status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	105 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	106
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	107 status = _cuda_cu_ctx_create(&_device_context, 0, cu_device);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	108 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	109 tty->print_cr("gpu_ptx::_cuda_cu_ctx_create(%x): %d", _device_context, status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	110 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	111
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	112 return status == 0; // CUDA_SUCCESS
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	113 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	114
10563 7a3499bf5e2c PTX kernel get_function return Morris Meyer <morris.meyer@oracle.com> parents: 9485 diff changeset	115 void gpu::Ptx::generate_kernel(unsigned char code, int code_len, const char *name) {
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	116
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	117 void *cu_module;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	118 const unsigned int jit_num_options = 3;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	119 int *jit_options = new int[jit_num_options];
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	120 void *jit_option_values = new void [jit_num_options];
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	121
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	122 jit_options[0] = 4; // CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	123 int jit_log_buffer_size = 1024;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	124 jit_option_values[0] = (void *)(size_t)jit_log_buffer_size;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	125
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	126 jit_options[1] = 3; // CU_JIT_INFO_LOG_BUFFER
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	127 char *jit_log_buffer = new char[jit_log_buffer_size];
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	128 jit_option_values[1] = jit_log_buffer;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	129
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	130 jit_options[2] = 0; // CU_JIT_MAX_REGISTERS
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	131 int jit_register_count = 32;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	132 jit_option_values[2] = (void *)(size_t)jit_register_count;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	133
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	134 int status = _cuda_cu_module_load_data_ex(&cu_module, code,
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	135 jit_num_options, jit_options, (void **)jit_option_values);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	136 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	137 tty->print_cr("gpu_ptx::_cuda_cu_module_load_data_ex(%x): %d", cu_module, status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	138 tty->print_cr("gpu_ptx::jit_log_buffer\n%s", jit_log_buffer);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	139 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	140
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	141 void *cu_function;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	142
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	143 status = _cuda_cu_module_get_function(&cu_function, cu_module, name);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	144 if (TraceWarpLoading) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	145 tty->print_cr("gpu_ptx::_cuda_cu_module_get_function(%s):%x %d", name, cu_function, status);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	146 }
10563 7a3499bf5e2c PTX kernel get_function return Morris Meyer <morris.meyer@oracle.com> parents: 9485 diff changeset	147 return cu_function;
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	148 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	149
10577 9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	150 bool gpu::Ptx::execute_kernel(address kernel) {
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	151 // grid dimensionality
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	152 unsigned int gridX = 1;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	153 unsigned int gridY = 1;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	154 unsigned int gridZ = 1;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	155
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	156 // thread dimensionality
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	157 unsigned int blockX = 1;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	158 unsigned int blockY = 1;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	159 unsigned int blockZ = 1;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	160
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	161 int cu_function = (int )kernel;
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	162
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	163 int status = _cuda_cu_launch_kernel(cu_function,
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	164 gridX, gridY, gridZ,
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	165 blockX, blockY, blockZ,
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	166 0, NULL, NULL, NULL);
10588 b0b368d38b40 CR-1032 - change ExternalCompilationResult kernel -> entryPoint Morris Meyer <morris.meyer@oracle.com> parents: 10577 diff changeset	167 if (TraceWarpLoading) {
b0b368d38b40 CR-1032 - change ExternalCompilationResult kernel -> entryPoint Morris Meyer <morris.meyer@oracle.com> parents: 10577 diff changeset	168 tty->print_cr("gpu_ptx::_cuda_cu_launch_kernel(%x): %d", kernel, status);
b0b368d38b40 CR-1032 - change ExternalCompilationResult kernel -> entryPoint Morris Meyer <morris.meyer@oracle.com> parents: 10577 diff changeset	169 }
10577 9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	170 return status == 0; // CUDA_SUCCESS
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	171 }
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	172
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	173 #ifdef __APPLE__
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	174 bool gpu::Ptx::probe_linkage_apple() {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	175 void *handle = dlopen("/usr/local/cuda/lib/libcuda.dylib", RTLD_LAZY);
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	176 if (handle != NULL) {
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	177 _cuda_cu_init =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	178 CAST_TO_FN_PTR(cuda_cu_init_func_t, dlsym(handle, "cuInit"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	179 _cuda_cu_ctx_create =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	180 CAST_TO_FN_PTR(cuda_cu_ctx_create_func_t, dlsym(handle, "cuCtxCreate"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	181 _cuda_cu_ctx_detach =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	182 CAST_TO_FN_PTR(cuda_cu_ctx_detach_func_t, dlsym(handle, "cuCtxDetach"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	183 _cuda_cu_ctx_synchronize =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	184 CAST_TO_FN_PTR(cuda_cu_ctx_synchronize_func_t, dlsym(handle, "cuCtxSynchronize"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	185 _cuda_cu_device_get_count =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	186 CAST_TO_FN_PTR(cuda_cu_device_get_count_func_t, dlsym(handle, "cuDeviceGetCount"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	187 _cuda_cu_device_get_name =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	188 CAST_TO_FN_PTR(cuda_cu_device_get_name_func_t, dlsym(handle, "cuDeviceGetName"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	189 _cuda_cu_device_get =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	190 CAST_TO_FN_PTR(cuda_cu_device_get_func_t, dlsym(handle, "cuDeviceGet"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	191 _cuda_cu_device_compute_capability =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	192 CAST_TO_FN_PTR(cuda_cu_device_compute_capability_func_t, dlsym(handle, "cuDeviceComputeCapability"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	193 _cuda_cu_module_get_function =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	194 CAST_TO_FN_PTR(cuda_cu_module_get_function_func_t, dlsym(handle, "cuModuleGetFunction"));
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	195 _cuda_cu_module_load_data_ex =
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	196 CAST_TO_FN_PTR(cuda_cu_module_load_data_ex_func_t, dlsym(handle, "cuModuleLoadDataEx"));
10577 9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	197 _cuda_cu_launch_kernel =
9c7d9e2c8326 PTX kernel execution - no args or return value Morris Meyer <morris.meyer@oracle.com> parents: 10566 diff changeset	198 CAST_TO_FN_PTR(cuda_cu_launch_kernel_func_t, dlsym(handle, "cuLaunchKernel"));
9430 147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	199 return true;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	200 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	201 return false;
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	202 }
147162b27799 GRAAL-234 - PTX code loading Morris Meyer <morris.meyer@oracle.com> parents: diff changeset	203 #endif

Mercurial > hg > truffle

annotate src/gpu/ptx/gpu_ptx.cpp @ 10806:f272fe8a2f9a