Unverified Commit 53643bdb by Zhi Committed by GitHub

[Fix] Fix CompilerAttrs (#5109)

* fix CompilerAttrs

* retrigger ci
parent 2e8f3a91
......@@ -143,9 +143,14 @@ class DebugAttrs(Attrs):
"""Attributes for debug"""
@tvm._ffi.register_object("relay.attrs.OnDeviceAttrs")
@tvm._ffi.register_object("relay.attrs.CompilerAttrs")
class CompilerAttrs(Attrs):
"""Attributes for compiler"""
@tvm._ffi.register_object("relay.attrs.DeviceCopyAttrs")
class DeviceCopyAttrs(Attrs):
"""Attributes for tensor.device_copy"""
"""Attributes for annotation.device_copy"""
@tvm._ffi.register_object("relay.attrs.CastAttrs")
......
......@@ -171,6 +171,8 @@ Mark a checkpoint for checkpointing memory optimization.
return outputs;
});
TVM_REGISTER_NODE_TYPE(CompilerAttrs);
RELAY_REGISTER_OP("annotation.compiler_begin")
.describe(R"code(
Beginning of a region that is handled by a given compiler.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment