H A D | gpu_backend_lib.cc | 337 Status NVPTXTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in NVPTXTargetModuleLinker() argument 371 Status LinkAndOptimizeModule(llvm::Module* module, GpuVersion gpu_version, in LinkAndOptimizeModule() argument 378 TF_RETURN_IF_ERROR(module_linker(module, gpu_version, hlo_module_config, in LinkAndOptimizeModule() 512 llvm::Module* module, GpuVersion gpu_version, in CompileToPtx() argument 536 std::get_if<se::CudaComputeCapability>(&gpu_version); in CompileToPtx() 556 module, gpu_version, hlo_module_config, libdevice_dir_path, in CompileToPtx() 788 Status AMDGPUTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in AMDGPUTargetModuleLinker() argument 794 std::get_if<se::RocmComputeCapability>(&gpu_version); in AMDGPUTargetModuleLinker() 861 llvm::Triple target_triple, GpuVersion gpu_version, in AMDGPUGetTargetMachine() argument 864 std::get_if<se::RocmComputeCapability>(&gpu_version); in AMDGPUGetTargetMachine() [all …]
|