Skip to content

Commit

Permalink
[XLA:GPU] Remove unused argument from TranslateLLVMToLLVMIR
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 686575649
  • Loading branch information
IllogicalMoose authored and Google-ML-Automation committed Oct 16, 2024
1 parent d7ca06c commit 6244044
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions xla/service/gpu/fusions/triton/triton_fusion_emitter.cc
Original file line number Diff line number Diff line change
Expand Up @@ -996,8 +996,7 @@ void LoadMlirDialectsForTriton(mlir::MLIRContext& mlir_context) {
// Simplified copy of translateLLVMToLLVMIR which in addition takes
// path to libdevice directly as an argument.
absl::StatusOr<std::unique_ptr<llvm::Module>> TranslateLLVMToLLVMIR(
llvm::LLVMContext* llvmContext, mlir::ModuleOp module,
absl::string_view libdevice_path) {
llvm::LLVMContext* llvmContext, mlir::ModuleOp module) {
mlir::DialectRegistry registry;
mlir::registerBuiltinDialectTranslation(registry);
mlir::registerLLVMDialectTranslation(registry);
Expand Down Expand Up @@ -1261,8 +1260,7 @@ absl::StatusOr<TritonWrapperResult> CompileTritonToLLVM(
if (emit_kernel) {
TF_ASSIGN_OR_RETURN(
std::unique_ptr<llvm::Module> ll_triton_module,
TranslateLLVMToLLVMIR(&llvm_module->getContext(), triton_module,
GetLibdevicePath(hlo_config, device_info)));
TranslateLLVMToLLVMIR(&llvm_module->getContext(), triton_module));
VLogModule(5, *ll_triton_module);
if (should_verify) {
VerifyModule(*ll_triton_module);
Expand Down

0 comments on commit 6244044

Please sign in to comment.