Searched refs:amdgpu_version (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| D | gpu_backend_lib.cc | 568 static std::vector<string> GetROCDLPaths(int amdgpu_version, in GetROCDLPaths() argument
594 absl::StrCat("oclc_isa_version_", amdgpu_version, ".bc"))); in GetROCDLPaths()
596 absl::StrCat("oclc_isa_version_", amdgpu_version, ".amdgcn.bc"))); in GetROCDLPaths()
776 Status LinkROCDLIfNecessary(llvm::Module* module, int amdgpu_version, in LinkROCDLIfNecessary() argument
783 GetROCDLPaths(amdgpu_version, rocdl_dir_path)); in LinkROCDLIfNecessary()
790 auto amdgpu_version = absl::get_if<std::pair<int, std::string>>(&gpu_version); in AMDGPUTargetModuleLinker() local
791 if (!amdgpu_version) { in AMDGPUTargetModuleLinker()
795 TF_RETURN_IF_ERROR(LinkROCDLIfNecessary(module, amdgpu_version->first, in AMDGPUTargetModuleLinker()
857 auto amdgpu_version = absl::get_if<std::pair<int, std::string>>(&gpu_version); in AMDGPUGetTargetMachine() local
858 int gcn_arch_value = amdgpu_version->first; in AMDGPUGetTargetMachine()
[all …]
|