Home
last modified time | relevance | path

Searched refs:gpus (Results 1 – 25 of 117) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig_test.py381 gpus = config.list_physical_devices('GPU')
382 self.assertGreater(len(gpus), 0)
437 gpus = config.list_physical_devices('GPU')
438 self.assertGreater(len(gpus), 0)
461 config.set_visible_devices(gpus)
468 gpus = config.list_physical_devices('GPU')
469 if len(gpus) < 2:
474 for i in range(0, len(gpus)):
480 with ops.device('/device:GPU:' + str(len(gpus))):
493 gpus = config.list_physical_devices('GPU')
[all …]
/aosp_15_r20/external/tensorflow/third_party/
H A Dcudnn_frontend_header_fix.patch32 +#include "third_party/gpus/cudnn/cudnn.h"
46 +#include "third_party/gpus/cudnn/cudnn.h"
47 +#include "third_party/gpus/cudnn/cudnn_backend.h"
61 +#include "third_party/gpus/cudnn/cudnn.h"
62 +#include "third_party/gpus/cudnn/cudnn_backend.h"
76 +#include "third_party/gpus/cudnn/cudnn.h"
77 +#include "third_party/gpus/cudnn/cudnn_backend.h"
90 +#include "third_party/gpus/cudnn/cudnn.h"
104 +#include "third_party/gpus/cudnn/cudnn.h"
105 +#include "third_party/gpus/cudnn/cudnn_backend.h"
[all …]
/aosp_15_r20/external/angle/src/gpu_info_util/
H A DSystemInfo.cpp80 for (const GPUDeviceInfo &gpu : gpus) in hasNVIDIAGPU()
92 for (const GPUDeviceInfo &gpu : gpus) in hasIntelGPU()
104 for (const GPUDeviceInfo &gpu : gpus) in hasAMDGPU()
119 for (size_t i = 0; i < gpus.size(); ++i) in getPreferredGPUIndex()
121 std::string vendor = VendorName(gpus[i].vendorId); in getPreferredGPUIndex()
261 ASSERT(!info->gpus.empty()); in GetDualGPUInfo()
271 for (size_t i = 0; i < info->gpus.size(); ++i) in GetDualGPUInfo()
273 if (IsIntel(info->gpus[i].vendorId)) in GetDualGPUInfo()
277 if (IsIntel(info->gpus[active].vendorId)) in GetDualGPUInfo()
285 info->isOptimus = hasIntel && IsNVIDIA(info->gpus[active].vendorId); in GetDualGPUInfo()
[all …]
H A DSystemInfo_linux.cpp74 if (!GetPCIDevicesWithLibPCI(&(info->gpus))) in GetSystemInfo()
84 if (info->gpus.size() == 0) in GetSystemInfo()
91 for (size_t i = 0; i < info->gpus.size(); ++i) in GetSystemInfo()
93 GPUDeviceInfo *gpu = &info->gpus[i]; in GetSystemInfo()
129 if (IsIntel(gpu->vendorId) && info->gpus.size() == 1) in GetSystemInfo()
140 info->gpus.emplace_back(std::move(nvidiaInfo)); in GetSystemInfo()
H A DSystemInfo_macos.mm158 if (info->gpus.size() < 2)
189 for (size_t i = 0; i < info->gpus.size(); ++i)
191 if (info->gpus[i].vendorId == activeVendor && info->gpus[i].deviceId == activeDevice)
326 GetIORegistryDevices(&info->gpus);
327 if (info->gpus.empty())
346 if (info->gpus.size() == 2 &&
347 ((IsIntel(info->gpus[0].vendorId) && !IsIntel(info->gpus[1].vendorId)) ||
348 (!IsIntel(info->gpus[0].vendorId) && IsIntel(info->gpus[1].vendorId))))
/aosp_15_r20/external/tensorflow/tensorflow/
H A Dopensource_only.files220 third_party/gpus/BUILD:
221 third_party/gpus/crosstool/BUILD.rocm.tpl:
222 third_party/gpus/crosstool/BUILD.tpl:
223 third_party/gpus/crosstool/BUILD:
224 third_party/gpus/crosstool/LICENSE:
225 third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc.tpl:
226 third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_rocm.tpl:
227 third_party/gpus/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl:
228 third_party/gpus/cuda/BUILD.tpl:
229 third_party/gpus/cuda/BUILD.windows.tpl:
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/tools/ci_build/
H A Dcuda-clang.patch1 diff --git a/third_party/gpus/crosstool/cc_toolchain_config.bzl.tpl b/third_party/gpus/crosstool/cc…
3 --- a/third_party/gpus/crosstool/cc_toolchain_config.bzl.tpl
4 +++ b/third_party/gpus/crosstool/cc_toolchain_config.bzl.tpl
14 diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl
16 --- a/third_party/gpus/cuda_configure.bzl
17 +++ b/third_party/gpus/cuda_configure.bzl
/aosp_15_r20/external/tensorflow/third_party/gpus/cuda/
H A DBUILD.tpl48 # Provides CUDA headers for '#include "third_party/gpus/cuda/include/cuda.h"'
56 include_prefix = "third_party/gpus",
88 include_prefix = "third_party/gpus/cuda/include",
97 include_prefix = "third_party/gpus/cuda/include",
106 include_prefix = "third_party/gpus/cuda/include",
115 include_prefix = "third_party/gpus/cuda/include",
124 include_prefix = "third_party/gpus/cuda/include",
162 include_prefix = "third_party/gpus/cudnn",
202 include_prefix = "third_party/gpus",
H A DBUILD.windows.tpl48 # Provides CUDA headers for '#include "third_party/gpus/cuda/include/cuda.h"'
56 include_prefix = "third_party/gpus",
89 include_prefix = "third_party/gpus/cuda/include",
98 include_prefix = "third_party/gpus/cuda/include",
107 include_prefix = "third_party/gpus/cuda/include",
116 include_prefix = "third_party/gpus/cuda/include",
125 include_prefix = "third_party/gpus/cuda/include",
158 include_prefix = "third_party/gpus/cudnn",
196 include_prefix = "third_party/gpus",
/aosp_15_r20/external/angle/src/tests/test_utils/
H A Dsystem_info_util.cpp24 if (systemInfo.gpus.size() < 2) in findGPU()
28 for (size_t i = 0; i < systemInfo.gpus.size(); ++i) in findGPU()
30 if (lowPower && IsIntel(systemInfo.gpus[i].vendorId)) in findGPU()
35 else if (!lowPower && !IsIntel(systemInfo.gpus[i].vendorId)) in findGPU()
61 for (size_t i = 0; i < systemInfo.gpus.size(); ++i) in FindActiveOpenGLGPU()
64 angle::SplitStringAlongWhitespace(VendorName(systemInfo.gpus[i].vendorId), &vendorTokens); in FindActiveOpenGLGPU()
H A Dangle_test_instantiate.cpp154 if (systemInfo->gpus.empty()) in GetActiveGPUDeviceInfo()
158 return &systemInfo->gpus[systemInfo->activeGPUIndex]; in GetActiveGPUDeviceInfo()
393 systemInfo.gpus.empty() ? 0 : systemInfo.gpus[systemInfo.activeGPUIndex].vendorId; in IsConfigAllowlisted()
/aosp_15_r20/hardware/google/gfxstream/host/gl/gl-host-common/opengl/
DNativeGpuInfo_windows.cpp264 static bool queryGpuInfoD3D(GpuInfoList* gpus) { in queryGpuInfoD3D() argument
281 gpus->addGpu(); in queryGpuInfoD3D()
282 GpuInfo& gpu = gpus->currGpu(); in queryGpuInfoD3D()
304 void getGpuInfoListNative(GpuInfoList* gpus) { in getGpuInfoListNative() argument
305 if (queryGpuInfoD3D(gpus)) return; in getGpuInfoListNative()
312 gpus->addGpu(); in getGpuInfoListNative()
313 GpuInfo& gpu = gpus->currGpu(); in getGpuInfoListNative()
319 Win32UnicodeString::convertToUtf8(device.DeviceID), gpus); in getGpuInfoListNative()
331 if (gpus->infos.empty()) { in getGpuInfoListNative()
/aosp_15_r20/prebuilts/cmake/linux-x86/share/cmake-3.22/Help/prop_test/
DRESOURCE_GROUPS.rst31 "2,gpus:2"
32 "gpus:4,crypto_chips:2")
35 semicolon.) The content of the first description is ``2,gpus:2``. This
37 GPU. The content of the second description is ``gpus:4,crypto_chips:2``. This
53 "gpus:1,gpus:1,gpus:1,gpus:1")
/aosp_15_r20/external/angle/src/feature_support_util/
H A Dfeature_support_util_unittest.cpp26 mSystemInfo.gpus.resize(1); in FeatureSupportUtilTest()
27 mSystemInfo.gpus[0].vendorId = 123; in FeatureSupportUtilTest()
28 mSystemInfo.gpus[0].deviceId = 234; in FeatureSupportUtilTest()
29 mSystemInfo.gpus[0].driverVendor = "GPUVendorA"; in FeatureSupportUtilTest()
30 mSystemInfo.gpus[0].detailedDriverVersion = {1, 2, 3, 4}; in FeatureSupportUtilTest()
153 systemInfo.gpus[0].detailedDriverVersion = {1, 2, 3, 5}; in TEST_F()
H A Dfeature_support_util.cpp812 systemInfo->gpus.resize(1); in ANGLEGetSystemInfo()
813 GPUDeviceInfo &gpu = systemInfo->gpus[0]; in ANGLEGetSystemInfo()
847 if (!rules || !systemInfo || !appName || (systemInfo->gpus.size() != 1)) in ANGLEShouldBeUsedForApplication()
854 Version gpuDriverVersion(systemInfo->gpus[0].detailedDriverVersion.major, in ANGLEShouldBeUsedForApplication()
855 systemInfo->gpus[0].detailedDriverVersion.minor, in ANGLEShouldBeUsedForApplication()
856 systemInfo->gpus[0].detailedDriverVersion.subMinor, in ANGLEShouldBeUsedForApplication()
857 systemInfo->gpus[0].detailedDriverVersion.patch); in ANGLEShouldBeUsedForApplication()
858 GPU gpuDriver(systemInfo->gpus[0].driverVendor, systemInfo->gpus[0].deviceId, in ANGLEShouldBeUsedForApplication()
/aosp_15_r20/external/mesa3d/src/etnaviv/drm-shim/
H A Detnaviv_noop.c42 static const struct etna_shim_gpu gpus[] = { variable
238 for (unsigned i = 0; i < ARRAY_SIZE(gpus); i++) { in drm_shim_driver_init()
239 if (strncasecmp(gpu, gpus[i].name, strlen(gpus[i].name)) == 0) { in drm_shim_driver_init()
240 shim_gpu = &gpus[i]; in drm_shim_driver_init()
247 shim_gpu = &gpus[1]; in drm_shim_driver_init()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Ddevice_placement_test.py120 gpus = config.list_physical_devices('GPU')
121 if not gpus:
137 gpus = config.list_physical_devices('GPU')
138 if not gpus:
253 gpus = config.list_physical_devices('GPU')
254 if not gpus:
268 gpus = config.list_physical_devices('GPU')
269 if not gpus:
/aosp_15_r20/external/angle/src/tests/
H A Dangle_system_info_tests_main.cpp104 if (info.gpus.empty()) in main()
130 js::Value gpus; in main() local
131 gpus.SetArray(); in main()
133 for (const angle::GPUDeviceInfo &gpu : info.gpus) in main()
161 gpus.PushBack(obj, allocator); in main()
164 doc.AddMember("gpus", gpus, allocator); in main()
/aosp_15_r20/external/mesa3d/src/panfrost/compiler/
H A Dcmdline.c101 } gpus[] = { in main() local
126 for (unsigned i = 0; i < ARRAY_SIZE(gpus); ++i) { in main()
127 if (strcmp(gpus[i].name, optarg)) in main()
130 unsigned major = gpus[i].major; in main()
131 unsigned minor = gpus[i].minor; in main()
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_common.py215 def __init__(self, gpus): argument
217 self.fc1 = nn.Linear(2, 10, bias=False).to(gpus[0])
218 self.fc2 = nn.Linear(10, 50, bias=False).to(gpus[1])
219 self.fc3 = nn.Linear(50, 4, bias=False).to(gpus[1])
223 ).to(gpus[0])
235 def __init__(self, gpus): argument
237 self.fc1 = nn.Linear(2, 10, bias=False).to(gpus[0])
238 self.fc2 = nn.Linear(10, 50, bias=False).to(gpus[1])
239 self.fc3 = nn.Linear(50, 4, bias=False).to(gpus[2])
240 self.fc4 = nn.Linear(4, 4, bias=False).to(gpus[3])
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/cuda/
H A DBUILD129 "//conditions:default": "//third_party/gpus/cuda:cublas_static",
157 "//conditions:default": "//third_party/gpus/cuda:cublas_lt_static",
194 "//conditions:default": "//third_party/gpus/cuda:cufft_static",
220 "//conditions:default": "//third_party/gpus/cudnn:cudnn",
255 "//conditions:default": "//third_party/gpus/cuda:curand_static",
287 "//conditions:default": "//third_party/gpus/cuda:cusolver_static",
303 "//conditions:default": "//third_party/gpus/cuda:cusparse_static",
/aosp_15_r20/external/vulkan-validation-layers/tests/
H A Dtest_environment.cpp121 ASSERT_LE(count, ARRAY_SIZE(gpus)); in SetUp()
122 err = vkEnumeratePhysicalDevices(inst, &count, gpus); in SetUp()
126 vk_testing::PhysicalDevice phys_dev(gpus[0]); in SetUp()
142 devs_.push_back(new Device(gpus[i])); in SetUp()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcompiled_collective_ops_gpu_test.py46 gpus = config.list_physical_devices('GPU')
47 if len(gpus) < num_gpus:
49 num_gpus, len(gpus)))
/aosp_15_r20/external/angle/src/tests/test_utils/third_party/
H A Dvulkan_command_buffer_utils.cpp139 res = vkEnumerateDeviceExtensionProperties(info.gpus[0], layer_name, in init_device_extension_properties()
151 res = vkEnumerateDeviceExtensionProperties(info.gpus[0], layer_name, in init_device_extension_properties()
271 info.gpus.resize(gpu_count); in init_enumerate_device()
273 res = vkEnumeratePhysicalDevices(info.inst, &gpu_count, info.gpus.data()); in init_enumerate_device()
276 vkGetPhysicalDeviceQueueFamilyProperties(info.gpus[0], &info.queue_family_count, NULL); in init_enumerate_device()
280 vkGetPhysicalDeviceQueueFamilyProperties(info.gpus[0], &info.queue_family_count, in init_enumerate_device()
285 vkGetPhysicalDeviceMemoryProperties(info.gpus[0], &info.memory_properties); in init_enumerate_device()
286 vkGetPhysicalDeviceProperties(info.gpus[0], &info.gpu_props); in init_enumerate_device()
621 vkGetPhysicalDeviceSurfaceSupportKHR(info.gpus[0], i, info.surface, &pSupportsPresent[i]); in init_swapchain_extension()
668 res = vkGetPhysicalDeviceSurfaceFormatsKHR(info.gpus[0], info.surface, &formatCount, NULL); in init_swapchain_extension()
[all …]
/aosp_15_r20/external/angle/src/tests/gl_tests/
H A DDrawBaseVertexBaseInstanceTest.cpp900 if (!systemInfo->gpus.empty()) in TEST_P()
902 ANGLE_SKIP_TEST_IF(0x6613 == systemInfo->gpus[systemInfo->activeGPUIndex].deviceId); in TEST_P()
937 if (!(systemInfo->activeGPUIndex < 0 || systemInfo->gpus.empty())) in TEST_P()
939 ANGLE_SKIP_TEST_IF(0x6613 == systemInfo->gpus[systemInfo->activeGPUIndex].deviceId); in TEST_P()
1089 if (!systemInfo->gpus.empty()) in TEST_P()
1091 ANGLE_SKIP_TEST_IF(0x6613 == systemInfo->gpus[systemInfo->activeGPUIndex].deviceId); in TEST_P()

12345