Mercurial > hg > truffle
annotate src/os/linux/vm/gpu_linux.cpp @ 13860:82090a107bae
make sure pushed values are formatted correctly
| author | Tom Rodriguez <tom.rodriguez@oracle.com> |
|---|---|
| date | Mon, 03 Feb 2014 17:16:52 -0800 |
| parents | 5c8a3c09397b |
| children | 51584f76462d |
| rev | line source |
|---|---|
| 10879 | 1 /* |
| 2 * Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved. | |
| 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. | |
| 4 * | |
| 5 * This code is free software; you can redistribute it and/or modify it | |
| 6 * under the terms of the GNU General Public License version 2 only, as | |
| 7 * published by the Free Software Foundation. | |
| 8 * | |
| 9 * This code is distributed in the hope that it will be useful, but WITHOUT | |
| 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| 12 * version 2 for more details (a copy is included in the LICENSE file that | |
| 13 * accompanied this code). | |
| 14 * | |
| 15 * You should have received a copy of the GNU General Public License version | |
| 16 * 2 along with this work; if not, write to the Free Software Foundation, | |
| 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | |
| 18 * | |
| 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA | |
| 20 * or visit www.oracle.com if you need additional information or have any | |
| 21 * questions. | |
| 22 * | |
| 23 */ | |
| 24 | |
| 25 #include "runtime/gpu.hpp" | |
| 26 #include "utilities/ostream.hpp" | |
| 27 | |
| 28 /* | |
| 29 * Probe for CUDA device on PCI bus using /proc/bus/pci/devices. Do | |
| 30 * not rely on CUDA tool kit being installed. We will check if CUDA | |
| 31 * library is installed later. | |
| 32 */ | |
| 33 | |
| 34 static unsigned int nvidia_vendor_id = 0x10de; | |
|
11606
4f69a5189e77
Create runtime object based on GPUIL type newly added to gpu class; some code refactoring.
bharadwaj
parents:
11485
diff
changeset
|
35 static unsigned int amd_vendor_id = 0x1002; |
| 10879 | 36 |
|
13828
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
37 #define PCI_DRIVER_NAME_START_POS 255 |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
38 |
|
13819
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
39 jobject gpu::probe_gpus(JNIEnv* env) { |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
40 bool hsail = false; |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
41 bool ptx = false; |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
42 |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
43 if (UseHSAILSimulator && gpu::Hsail::register_natives(env)) { |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
44 hsail = true; |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
45 } |
|
12743
f1a55428a8d7
more HSAIL support in the C++ layer for executing HSAIL code on the simulator
Doug Simon <doug.simon@oracle.com>
parents:
11606
diff
changeset
|
46 |
|
f1a55428a8d7
more HSAIL support in the C++ layer for executing HSAIL code on the simulator
Doug Simon <doug.simon@oracle.com>
parents:
11606
diff
changeset
|
47 /* |
|
11606
4f69a5189e77
Create runtime object based on GPUIL type newly added to gpu class; some code refactoring.
bharadwaj
parents:
11485
diff
changeset
|
48 * Open /proc/bus/pci/devices to look for the first GPU device. For |
|
4f69a5189e77
Create runtime object based on GPUIL type newly added to gpu class; some code refactoring.
bharadwaj
parents:
11485
diff
changeset
|
49 * now, we will just find the first GPU device. Will need to revisit |
|
4f69a5189e77
Create runtime object based on GPUIL type newly added to gpu class; some code refactoring.
bharadwaj
parents:
11485
diff
changeset
|
50 * this to support execution on multiple GPU devices, if they exist. |
|
11292
5040ec3ff3aa
Defer PTX on Mac probe to the CUDA driver initialization call
Morris Meyer <morris.meyer@oracle.com>
parents:
10879
diff
changeset
|
51 */ |
| 10879 | 52 FILE *pci_devices = fopen("/proc/bus/pci/devices", "r"); |
| 53 char contents[4096]; | |
| 54 unsigned int bus_num_devfn_ign; | |
| 55 unsigned int vendor; | |
| 56 unsigned int device; | |
|
13828
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
57 const char *driver_name_string = "nvidia"; |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
58 const int driver_name_string_len = strlen(driver_name_string); |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
59 |
| 10879 | 60 if (pci_devices == NULL) { |
| 61 tty->print_cr("*** Failed to open /proc/bus/pci/devices"); | |
|
13819
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
62 return NULL; |
| 10879 | 63 } |
| 64 | |
| 65 while (fgets(contents, sizeof(contents)-1, pci_devices)) { | |
| 66 sscanf(contents, "%04x%04x%04x", &bus_num_devfn_ign, &vendor, &device); | |
|
11485
49bb1bc983c6
Implement several missing PTX codegen features; return value capture and method args passing of java method executed on GPU.
bharadwaj
parents:
11292
diff
changeset
|
67 if (vendor == nvidia_vendor_id) { |
|
13828
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
68 /* Check if this device is registered to be using nvidia driver */ |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
69 if (strncmp(&contents[PCI_DRIVER_NAME_START_POS], |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
70 driver_name_string, driver_name_string_len) == 0) { |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
71 if (TraceGPUInteraction) { |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
72 tty->print_cr("Found supported nVidia device [vendor=0x%04x, device=0x%04x]", vendor, device); |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
73 } |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
74 if (!ptx && gpu::Ptx::register_natives(env)) { |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
75 ptx = true; |
|
5c8a3c09397b
Add additional device checks and compute capability checks for CUDA devices found on Linux.
S.Bharadwaj Yadavalli <bharadwaj.yadavalli@oracle.com>
parents:
13819
diff
changeset
|
76 } |
|
13819
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
77 } |
| 10879 | 78 } |
| 79 } | |
| 80 | |
| 81 // Close file pointer. | |
| 82 fclose(pci_devices); | |
| 83 | |
|
13819
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
84 const char* gpus = ""; |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
85 if (ptx && hsail) { |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
86 gpus = "PTX,HSAIL"; |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
87 } else if (ptx) { |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
88 gpus = "PTX"; |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
89 } else if (hsail) { |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
90 gpus = "HSAIL"; |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
91 } |
|
49db2c1e3bee
added support for co-existing GPU backends (JBS:GRAAL-1)
Doug Simon <doug.simon@oracle.com>
parents:
13114
diff
changeset
|
92 return env->NewStringUTF(gpus); |
| 10879 | 93 } |