mirror of https://github.com/llvm/torch-mlir
40 lines
1.3 KiB
C++
40 lines
1.3 KiB
C++
//===- generic.h ----------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, 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.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
// This file is adapted from pytorch/pytorch
|
|
// https://github.com/pytorch/pytorch/blob/master/torch/csrc/lazy/ts_backend/generic.h
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#pragma once
|
|
|
|
#include "../mlir_node.h"
|
|
|
|
namespace torch {
|
|
namespace lazy {
|
|
|
|
// Generic IR Node implementation for nodes which can simply be described by a
|
|
// specific OpKind and a lowering function. IR nodes carrying
|
|
// metadata should not be using this class TORCH_API (and have the metadata
|
|
// captured by the LowerFn), but they should instead create a dedicated IR node.
|
|
// Doing the former would limit IR introspection.
|
|
class TORCH_API Generic : public TorchMlirNode {
|
|
public:
|
|
Generic(
|
|
OpKind op,
|
|
OpList operands,
|
|
Shape shape,
|
|
size_t num_outputs = 1,
|
|
hash_t hash_seed = static_cast<uint32_t>(0x5a2d296e9));
|
|
|
|
private:
|
|
hash_t hash_seed_;
|
|
};
|
|
|
|
} // namespace lazy
|
|
} // namespace torch
|