#pragma once
|
|
#include <c10/macros/Export.h>
|
|
namespace c10 {
|
struct OperatorName;
|
}
|
|
namespace at {
|
|
/*
|
There's semantically three sets of operators:
|
|
- aten_ops_already_moved_to_c10
|
- aten_ops_not_moved_to_c10_yet
|
- non_aten_ops (e.g. custom ops)
|
|
register_c10_ops.cpp needs to decide between aten_ops_already_moved_to_c10
|
and union(aten_ops_not_moved_to_c10_yet, non_aten_ops).
|
The c10 operator registry needs to decide between aten_ops_not_moved_to_c10_yet
|
and union(aten_ops_already_moved_to_c10, non_aten_ops), which is different to what
|
register_c10_ops.cpp needs. We need to store two sets to be able to make both decisions.
|
*/
|
|
// list of ATen ops that got already moved to the c10 dispatcher
|
CAFFE2_API bool aten_op_is_already_moved_to_c10(const c10::OperatorName& opName);
|
|
// list of ATen ops that are still on the globalATenDispatch dispatcher.
|
CAFFE2_API bool aten_op_is_not_moved_to_c10_yet(const c10::OperatorName& opName);
|
|
}
|
