mirror of https://github.com/llvm/torch-mlir
Fix update_torch_ods.sh
parent
5b6902e31c
commit
f0ed9e2d8d
|
@ -2,10 +2,10 @@
|
|||
# Updates auto-generated ODS files for the `torch` dialect.
|
||||
set -e
|
||||
|
||||
src_dir="$(realpath $(dirname $0)/../../..)"
|
||||
src_dir="$(realpath $(dirname $0)/..)"
|
||||
build_dir="$(realpath "${TORCH_MLIR_BUILD_DIR:-$src_dir/build}")"
|
||||
torch_ir_dir="${src_dir}/external/torch-mlir/include/torch-mlir/Dialect/Torch/IR"
|
||||
python_packages_dir="${build_dir}/python_packages"
|
||||
torch_ir_dir="${src_dir}/include/torch-mlir/Dialect/Torch/IR"
|
||||
python_packages_dir="${build_dir}/tools/torch-mlir/python_packages"
|
||||
|
||||
#ninja -C "${build_dir}"
|
||||
PYTHONPATH="${python_packages_dir}/torch_mlir" python \
|
||||
|
|
|
@ -2438,3 +2438,4 @@ def Torch_AtenEqDeviceOp : Torch_Op<"aten.eq.device", [
|
|||
);
|
||||
let assemblyFormat = "$a `,` $b attr-dict `:` type($a) `,` type($b) `->` type($result)";
|
||||
}
|
||||
|
||||
|
|
|
@ -224,3 +224,4 @@ def Torch_PrimTolistOp : Torch_Op<"prim.tolist", [
|
|||
);
|
||||
let assemblyFormat = "`(` $operands `)` attr-dict `:` type($operands) `->` type($results)";
|
||||
}
|
||||
|
||||
|
|
|
@ -33,3 +33,4 @@ def Torch_QuantizedLinearOp : Torch_Op<"quantized.linear", [
|
|||
);
|
||||
let assemblyFormat = "$X `,` $W_prepack `,` $Y_scale_i `,` $Y_zero_point_i attr-dict `:` type($X) `,` type($W_prepack) `,` type($Y_scale_i) `,` type($Y_zero_point_i) `->` type($Y)";
|
||||
}
|
||||
|
||||
|
|
|
@ -265,8 +265,8 @@ ODS_BANNER = "\n".join([
|
|||
"//",
|
||||
"// This file is licensed under the Apache License v2.0 with LLVM Exceptions.",
|
||||
"// See https://llvm.org/LICENSE.txt for license information.",
|
||||
"// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||||
// Also available under a BSD-style license. See LICENSE.",
|
||||
"// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception",
|
||||
"// Also available under a BSD-style license. See LICENSE.",
|
||||
"//",
|
||||
"// Operation summaries and descriptions were systematically derived from public",
|
||||
"// API docstrings and are licensed accordingly:",
|
||||
|
|
Loading…
Reference in New Issue