Skip to content

Commit

Permalink
[ROCm] Fix build break in xla/service/gpu/ir_emitter_triton_rocm.cc
Browse files Browse the repository at this point in the history
  • Loading branch information
@zoranjovanovic-ns @Ruturaj4
zoranjovanovic-ns authored and Ruturaj4 committed Jun 4, 2024
1 parent 312c75a commit 73d1584
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions xla/service/gpu/ir_emitter_triton_rocm.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -79,8 +79,8 @@ absl::Status CreateTritonPipeline(
// TODO ROCm Check if we want to compare MI100 and greater
pm.addPass(mt::gpu::createTritonGPUOptimizeDotOperands({true}));
pm.addPass(mlir::createCSEPass());
pm.addPass(mt::gpu::createTritonGPUPipeline(
{config.num_stages, config.num_warps, config.num_ctas, ccAsInt}));
pm.addPass(mt::gpu::createTritonGPUPipeline({config.num_stages, config.num_warps,
config.num_ctas, ccAsInt}));
pm.addPass(mt::gpu::createTritonGPUPrefetch());

// TODO ROCm Check if we want to compare MI100 and greater
Expand Down

1 comment on commit 73d1584

@i-chaochen
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

are we going to upstream this change?

Please sign in to comment.