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): ...@@ -143,9 +143,14 @@ class DebugAttrs(Attrs):
"""Attributes for debug""" """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): class DeviceCopyAttrs(Attrs):
"""Attributes for tensor.device_copy""" """Attributes for annotation.device_copy"""
@tvm._ffi.register_object("relay.attrs.CastAttrs") @tvm._ffi.register_object("relay.attrs.CastAttrs")
......
...@@ -171,6 +171,8 @@ Mark a checkpoint for checkpointing memory optimization. ...@@ -171,6 +171,8 @@ Mark a checkpoint for checkpointing memory optimization.
return outputs; return outputs;
}); });
TVM_REGISTER_NODE_TYPE(CompilerAttrs);
RELAY_REGISTER_OP("annotation.compiler_begin") RELAY_REGISTER_OP("annotation.compiler_begin")
.describe(R"code( .describe(R"code(
Beginning of a region that is handled by a given compiler. 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