Home
last modified time | relevance | path

Searched refs:cuda_compute_capability (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc56 absl::optional<CudaComputeCapability> cuda_compute_capability = in CompileMlirModule() local
58 CudaComputeCapability cuda_compute_capability; in CompileMlirModule() local
59 stream_exec->GetDeviceDescription().cuda_compute_capability( in CompileMlirModule()
60 &cuda_compute_capability.cc_major, &cuda_compute_capability.cc_minor); in CompileMlirModule()
61 if (cuda_compute_capability.cc_major == -1) { in CompileMlirModule()
64 return cuda_compute_capability; in CompileMlirModule()
69 backend_->platform()->Name(), gpu_device_info, cuda_compute_capability, in CompileMlirModule()
Dhlo_to_llvm_ir.cc53 xla::gpu::CudaComputeCapability cuda_compute_capability; in CompileAndPrintLlvmIr() local
54 cuda_compute_capability.cc_major = 7; in CompileAndPrintLlvmIr()
55 cuda_compute_capability.cc_minor = 0; in CompileAndPrintLlvmIr()
64 cuda_compute_capability, /*pointer_size=*/8)); in CompileAndPrintLlvmIr()
Dreduction_vectorization_test.cc57 executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in TEST_F()
105 executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in TEST_F()
153 executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in TEST_F()
237 executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in TEST_F()
342 device_description.cuda_compute_capability(&cc_major, &cc_minor); in TEST_F()
Dgpu_atomic_test.cc115 device_description.cuda_compute_capability(&cc_major, &cc_minor); in TEST_F()
/external/tensorflow/tensorflow/tools/tensorflow_builder/config_detector/
DBUILD20 ":cuda_compute_capability",
28 name = "cuda_compute_capability",
29 srcs = ["data/cuda_compute_capability.py"],
Dconfig_detector.py77 from tensorflow.tools.tensorflow_builder.config_detector.data import cuda_compute_capability
255 gpu_dict = cuda_compute_capability.retrieve_from_golden()
394 cuda_compute_capa = cuda_compute_capability.retrieve_from_web()
396 cuda_compute_capa = cuda_compute_capability.retrieve_from_golden()
/external/tensorflow/tensorflow/core/grappler/
Ddevices.cc54 desc->cuda_compute_capability(&cc_major, &cc_minor); in GetNumAvailableGPUs()
55 std::pair<int, int> cuda_compute_capability(cc_major, cc_minor); in GetNumAvailableGPUs() local
58 cuda_compute_capability >= min_cuda_compute_capability) { in GetNumAvailableGPUs()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h43 absl::optional<CudaComputeCapability> cuda_compute_capability, in IrEmitterContext() argument
50 cuda_compute_capability_(cuda_compute_capability), in IrEmitterContext()
65 absl::optional<CudaComputeCapability> cuda_compute_capability() const { in cuda_compute_capability() function
Dgpu_compiler.cc552 absl::optional<CudaComputeCapability> cuda_compute_capability, in CompileModuleToLlvmIrImpl() argument
598 cuda_compute_capability, profile_index_map, &mlir_context, in CompileModuleToLlvmIrImpl()
869 absl::optional<CudaComputeCapability> cuda_compute_capability = in RunBackend() local
871 CudaComputeCapability cuda_compute_capability; in RunBackend() local
872 stream_exec->GetDeviceDescription().cuda_compute_capability( in RunBackend()
873 &cuda_compute_capability.cc_major, &cuda_compute_capability.cc_minor); in RunBackend()
874 if (cuda_compute_capability.cc_major == -1) { in RunBackend()
877 return cuda_compute_capability; in RunBackend()
906 stream_exec->platform()->Name(), gpu_device_info, cuda_compute_capability, in RunBackend()
987 absl::optional<CudaComputeCapability> cuda_compute_capability, in CompileModuleToLlvmIr() argument
[all …]
Dir_emission_utils.h204 absl::optional<CudaComputeCapability> cuda_compute_capability);
Dgpu_compiler.h157 absl::optional<CudaComputeCapability> cuda_compute_capability,
Dir_emission_utils.cc136 absl::optional<CudaComputeCapability> cuda_compute_capability) { in GetReductionTiling() argument
148 if (cuda_compute_capability) { in GetReductionTiling()
149 cc_major = cuda_compute_capability->cc_major; in GetReductionTiling()
Dnvptx_compiler.cc289 if (!stream_exec->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetGpuVersion()
Dstream_executor_util.cc43 CHECK(stream_executor.GetDeviceDescription().cuda_compute_capability(&major, in IsVoltaOrLater()
Dir_emitter.cc295 ir_emitter_context_->cuda_compute_capability()->cc_major >= 6; in MaybeEmitDirectAtomicOperation()
Dgpu_executable.cc111 main_stream->parent()->GetDeviceDescription().cuda_compute_capability( in CheckCompatibilityWithServiceExecutableRunOptions()
Dgpu_conv_algorithm_picker.cc182 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc114 bool DeviceDescription::cuda_compute_capability(int *major, int *minor) const { in cuda_compute_capability() function in stream_executor::DeviceDescription
Ddevice_description.h134 bool cuda_compute_capability(int *major, int *minor) const;
Dstream_executor_pimpl.cc272 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveAlgorithms()
313 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveBackwardDataAlgorithms()
326 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveBackwardFilterAlgorithms()
/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.cc135 if (description.cuda_compute_capability(&major_version, &minor_version)) { in IsDeviceSupported()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h38 .cuda_compute_capability(&major, &minor)); in IsVoltaOrLater()
Dgpu_utils.cc117 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc972 if (!se->GetDeviceDescription().cuda_compute_capability(&cc_major, in SingleVirtualDeviceMemoryLimit()
1136 if (desc->cuda_compute_capability(&cc_major, &cc_minor)) { in GetDeviceDetails()
1384 if (!desc.cuda_compute_capability(&cc_major, &cc_minor)) { in GetShortDeviceDescription()
1765 if (!description->cuda_compute_capability(&cc_major, &cc_minor)) { in GetValidDeviceIds()
1851 if (!desc->cuda_compute_capability(&device_capability.major_part, in GetValidDeviceIds()
Dgpu_device_test.cc48 if (!se->GetDeviceDescription().cuda_compute_capability(cc_major, cc_minor)) { in GetComputeCapability()

12