summaryrefslogtreecommitdiff
path: root/compiler/loco/include/loco/IR/NodeMixins.h
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/loco/include/loco/IR/NodeMixins.h')
-rw-r--r--compiler/loco/include/loco/IR/NodeMixins.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/compiler/loco/include/loco/IR/NodeMixins.h b/compiler/loco/include/loco/IR/NodeMixins.h
index f0e34b0ba..fcadcaba7 100644
--- a/compiler/loco/include/loco/IR/NodeMixins.h
+++ b/compiler/loco/include/loco/IR/NodeMixins.h
@@ -83,7 +83,7 @@ private:
std::vector<Dimension> _dims;
};
-template <unsigned N> struct FixedArity
+template <uint32_t N> struct FixedArity
{
template <typename Base> class Mixin : public virtual Base
{
@@ -99,7 +99,7 @@ template <unsigned N> struct FixedArity
virtual ~Mixin() = default;
public:
- unsigned arity(void) const final { return N; }
+ uint32_t arity(void) const final { return N; }
Node *arg(uint32_t n) const final { return _args.at(n)->node(); }
@@ -113,7 +113,7 @@ template <unsigned N> struct FixedArity
protected:
// This API allows inherited classes to access "_args" field.
- Use *at(unsigned n) const { return _args.at(n).get(); }
+ Use *at(uint32_t n) const { return _args.at(n).get(); }
private:
std::array<std::unique_ptr<Use>, N> _args{};