MLIR
20.0.0git
lib
Dialect
Tensor
Extensions
AllExtensions.cpp
Go to the documentation of this file.
1
//===- AllExtensions.cpp - All Tensor Dialect Extensions ------------------===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8
9
#include "
mlir/Dialect/Tensor/Extensions/AllExtensions.h
"
10
#include "
mlir/Dialect/Tensor/Extensions/MeshShardingExtensions.h
"
11
12
using namespace
mlir
;
13
14
void
mlir::tensor::registerAllExtensions
(
DialectRegistry
®istry) {
15
registerShardingInterfaceExternalModels
(registry);
16
}
AllExtensions.h
MeshShardingExtensions.h
mlir::DialectRegistry
The DialectRegistry maps a dialect namespace to a constructor for the matching dialect.
Definition:
DialectRegistry.h:139
mlir::func::registerShardingInterfaceExternalModels
void registerShardingInterfaceExternalModels(DialectRegistry ®istry)
Definition:
MeshShardingExtensions.cpp:16
mlir::tensor::registerAllExtensions
void registerAllExtensions(DialectRegistry ®istry)
Register all extensions of the Tensor dialect.
Definition:
AllExtensions.cpp:14
mlir
Include the generated interface declarations.
Definition:
LocalAliasAnalysis.h:20
Generated on Tue Dec 24 2024 00:40:04 for MLIR by
1.9.1