mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
15 lines
432 B
C++
15 lines
432 B
C++
#include "triton/Dialect/TritonGPU/IR/Traits.h"
|
|
#include "triton/Dialect/TritonGPU/IR/Dialect.h"
|
|
|
|
mlir::LogicalResult
|
|
mlir::OpTrait::impl::verifyResultsAreSharedEncoding(Operation *op) {
|
|
if (failed(verifyAtLeastNResults(op, 1)))
|
|
return failure();
|
|
|
|
for (auto result : op->getResults())
|
|
if (!isSharedEncoding(result))
|
|
return op->emitOpError() << "requires all results to be shared encoding";
|
|
|
|
return success();
|
|
};
|