Skip to content

Commit

Permalink
Fix format errors.
Browse files Browse the repository at this point in the history
  • Loading branch information
JamesTheZ committed Jul 24, 2023
1 parent bb47360 commit 3b3f9fc
Showing 1 changed file with 13 additions and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,10 @@ using namespace mlir::torch::TorchConversion;

namespace {

static std::vector<std::string> quantizationOpList{"torch_blade.fake_quant",
"torch_blade.quantize",
"torch_blade.dequantize"};
static std::vector<std::string> quantizationOpList{
"torch_blade.fake_quant",
"torch_blade.quantize",
"torch_blade.dequantize"};
static std::string customCallName = "torch_blade.custom_call";

class ConvertOperatorOp : public OpConversionPattern<OperatorOp> {
Expand Down Expand Up @@ -184,14 +185,15 @@ class ConvertOperatorOp : public OpConversionPattern<OperatorOp> {
resultTypes.push_back(resultTy);
}

const std::vector<std::string> requiredAttrName{"call_target_name",
"device",
"input_placements",
"output_placements",
"input_layouts",
"output_layouts",
"expected_input_layouts",
"expected_output_layouts"};
const std::vector<std::string> requiredAttrName{
"call_target_name",
"device",
"input_placements",
"output_placements",
"input_layouts",
"output_layouts",
"expected_input_layouts",
"expected_output_layouts"};

for (const auto& n : requiredAttrName) {
if (!op->hasAttr(n)) {
Expand Down

0 comments on commit 3b3f9fc

Please sign in to comment.