Skip to content

Commit 01230fa

Browse files
committed
Fix CI
1 parent 359bf00 commit 01230fa

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/tir/schedule/primitive/annotate.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,8 @@ struct AnnotateTraits : public UnpackedInstTraits<AnnotateTraits> {
127127
return py.Str();
128128
}
129129

130-
friend struct UnpackedInstTraits;
130+
template <typename>
131+
friend struct ::tvm::tir::UnpackedInstTraits;
131132
};
132133

133134
struct UnannotateTraits : public UnpackedInstTraits<UnannotateTraits> {
@@ -158,7 +159,8 @@ struct UnannotateTraits : public UnpackedInstTraits<UnannotateTraits> {
158159
return py.Str();
159160
}
160161

161-
friend struct UnpackedInstTraits;
162+
template <typename>
163+
friend struct ::tvm::tir::UnpackedInstTraits;
162164
};
163165

164166
TVM_REGISTER_INST_KIND_TRAITS(AnnotateTraits);

0 commit comments

Comments
 (0)