Home
last modified time | relevance | path

Searched refs:kTargetTriple (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_constants.h24 constexpr char kTargetTriple[] = "nvptx64-nvidia-cuda"; variable
34 constexpr char kTargetTriple[] = "amdgcn-amd-amdhsa"; variable
Damdgpu_compiler.cc92 : GpuCompiler(stream_executor::rocm::kROCmPlatformId, amdgpu::kTargetTriple, in AMDGPUCompiler()
Dnvptx_compiler.cc280 : GpuCompiler(stream_executor::cuda::kCudaPlatformId, nvptx::kTargetTriple, in NVPTXCompiler()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc46 llvm_module->setTargetTriple(amdgpu::kTargetTriple); in CompileMlirModule()
49 llvm_module->setTargetTriple(nvptx::kTargetTriple); in CompileMlirModule()
Dhlo_to_llvm_ir.cc61 /*target_triple=*/xla::gpu::nvptx::kTargetTriple, in CompileAndPrintLlvmIr()