wehub-resource-sync
103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
# isort: skip_file
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
"""Relax transformations."""
|
|
|
|
from .transform import (
|
|
AdjustMatmulOrder,
|
|
AllocateWorkspace,
|
|
AlterOpImpl,
|
|
AnnotateTIROpPattern,
|
|
AttachAttrLayoutFreeBuffers,
|
|
AttachGlobalSymbol,
|
|
BindParams,
|
|
BindSymbolicVars,
|
|
BundleModelParams,
|
|
CallTIRRewrite,
|
|
CanonicalizeBindings,
|
|
CombineParallelMatmul,
|
|
ComputePrimValue,
|
|
ConvertLayout,
|
|
ConvertToDataflow,
|
|
DataflowBlockPass,
|
|
DataflowUseInplaceCalls,
|
|
DeadCodeElimination,
|
|
DecomposeOpsForInference,
|
|
DecomposeOpsForTraining,
|
|
EliminateCommonSubexpr,
|
|
ExpandMatmulOfSum,
|
|
ExpandTupleArguments,
|
|
FoldConstant,
|
|
FunctionPass,
|
|
FuseOps,
|
|
FuseOpsByPattern,
|
|
FuseTIR,
|
|
FusionPattern,
|
|
Gradient,
|
|
InlinePrivateFunctions,
|
|
KillAfterLastUse,
|
|
LambdaLift,
|
|
LazyGetInput,
|
|
LazySetOutput,
|
|
LegalizeOps,
|
|
LiftTransformParams,
|
|
LowerAllocTensor,
|
|
LowerRuntimeBuiltin,
|
|
MergeCompositeFunctions,
|
|
MetaScheduleApplyDatabase,
|
|
MetaScheduleTuneIRMod,
|
|
MetaScheduleTuneTIR,
|
|
Normalize,
|
|
NormalizeGlobalVar,
|
|
PatternCheckContext,
|
|
RealizeVDevice,
|
|
RemovePurityChecking,
|
|
RemoveUnusedOutputs,
|
|
RemoveUnusedParameters,
|
|
ReorderPermuteDimsAfterConcat,
|
|
ReorderTakeAfterMatmul,
|
|
RewriteCUDAGraph,
|
|
RewriteDataflowReshape,
|
|
RunCodegen,
|
|
SplitCallTIRByPattern,
|
|
SplitLayoutRewritePreproc,
|
|
StaticPlanBlockMemory,
|
|
ToMixedPrecision,
|
|
ToNonDataflow,
|
|
TopologicalSort,
|
|
UpdateParamType,
|
|
UpdateVDevice,
|
|
VMBuiltinLower,
|
|
VMShapeLower,
|
|
SpecializePrimFuncBasedOnCallSite,
|
|
dataflowblock_pass,
|
|
function_pass,
|
|
)
|
|
|
|
from .attach_external_modules import AttachExternModules
|
|
from .fast_math import FastMathTransform
|
|
from .fuse_transpose_matmul import FuseTransposeMatmul
|
|
from .ipc_allreduce_rewrite import IPCAllReduceRewrite
|
|
from .lazy_transform_params import LazyTransformParams
|
|
from .lower_gpu_ipc_alloc_storage import LowerGPUIPCAllocStorage
|
|
from .optimize_layout_transform import OptimizeLayoutTransform
|
|
from .fold_batch_norm_to_conv2d_for_inference import FoldBatchnormToConv2D
|
|
from .remove_redundant_reshape import RemoveRedundantReshape
|
|
|
|
# Import to register the legalization functions.
|
|
from . import legalize_ops
|