mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
17 lines
547 B
C++
17 lines
547 B
C++
#ifndef TRITON_CONVERSION_TRITONGPU_TO_ELEMENTWISE_OP_H
|
|
#define TRITON_CONVERSION_TRITONGPU_TO_ELEMENTWISE_OP_H
|
|
|
|
#include "TritonGPUToLLVMBase.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::triton;
|
|
|
|
void populateElementwiseOpToLLVMPatterns(
|
|
TritonGPUToLLVMTypeConverter &typeConverter, RewritePatternSet &patterns,
|
|
int numWarps, ModuleAxisInfoAnalysis &axisInfoAnalysis,
|
|
ModuleAllocation &allocation,
|
|
ConvertTritonGPUOpToLLVMPatternBase::IndexCacheInfo &indexCacheInfo,
|
|
int computeCapability, PatternBenefit benefit);
|
|
|
|
#endif
|