Skip to content

Commit

Permalink
Revert "[TorchToTosa] Refactoring to separate construction of legal/i…
Browse files Browse the repository at this point in the history
…llegal ops and conversion patterns. (#3759)"

This reverts commit f03a576.
  • Loading branch information
rahuls-cerebras committed Jan 3, 2025
1 parent c6c429f commit 51ec34c
Show file tree
Hide file tree
Showing 2 changed files with 222 additions and 249 deletions.
15 changes: 1 addition & 14 deletions include/torch-mlir/Conversion/TorchToTosa/TorchToTosa.h
Original file line number Diff line number Diff line change
Expand Up @@ -12,25 +12,12 @@

#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Transforms/DialectConversion.h"

#include <memory>

namespace mlir {
namespace torch {

/// Collect a set of legal/illegal ops for converting Torch operations to Tosa
/// dialect.
void populateTorchToTosaConversionLegalOps(ConversionTarget &target);

/// Collect a set of patterns to convert Torch operations to Tosa dialect +
/// return the set of illegalOps
std::set<StringRef>
populateTorchToTosaConversionPatternsAndIllegalOps(TypeConverter &typeConverter,
RewritePatternSet &patterns);

std::unique_ptr<OperationPass<func::FuncOp>> createConvertTorchToTosaPass();
} // namespace torch
}
} // namespace mlir

#endif // TORCHMLIR_CONVERSION_TORCHTOTOSA_TORCHTOTOSA_H
Loading

0 comments on commit 51ec34c

Please sign in to comment.