[PATCH] Add iterator_adaptor to iterate over SDNode operand SDNode's
Pete Cooper
peter_cooper at apple.com
Thu Jun 25 17:36:41 PDT 2015
> On Jun 25, 2015, at 5:21 PM, David Blaikie <dblaikie at gmail.com> wrote:
>
>
>
> On Thu, Jun 25, 2015 at 5:11 PM, Pete Cooper <peter_cooper at apple.com <mailto:peter_cooper at apple.com>> wrote:
>
>> On Jun 25, 2015, at 2:40 PM, David Blaikie <dblaikie at gmail.com <mailto:dblaikie at gmail.com>> wrote:
>>
>>
>>
>> On Thu, Jun 25, 2015 at 2:30 PM, Pete Cooper <peter_cooper at apple.com <mailto:peter_cooper at apple.com>> wrote:
>> Hi David
>>
>> Currently the only SDNode iterator over operands does so with SDUse*. Users frequently then call getNode() on the operand.
>>
>> This patch adds an iterator to SDNode which returns the SDNode of the operand. This allows more patterns to be converted to foreach. It is based on value_op_iterator which I found in User.h.
>>
>> For now i’ve only used it in a single place, but I found a bunch more in DAGCombiner for example which should be applicable. I would convert those in a later commit assuming you are ok with this solution.
>>
>> This immediately raises red flags:
>>
>> SDNode *operator*() const { return I->getNode(); }
>> SDNode *operator->() const { return operator*(); }
>>
>> op* should return a T& and op-> should return T*
> I’d forgotten about that.
>>
>> If these SDNode*s can never be null, then perhaps this should be:
> I wasn’t actually sure if they could be. My initial reaction was that null operands wouldn’t make sense, but it turns out we never checked. So here’s a patch which does actually ensure that the SDNode's referenced as operands are never null. It passes make check. I can put it on another email for review if you prefer I don’t add it here.
>
> It might be worth a separate thread, or at least a drive-by by someone who deals with this part of the code. I don't really understand the necessity/merits/drawbacks of the 'SDUse::reset()" member function you've introduced.
No problem. Thanks for taking a look. I’ve just sent out an email to llvm-commits and asked Hal for review as I know he’s done lots of SD work.
>
>
>
>>
>> SDNode &operator*
>> SDNode *operator->
>>
>> ? (because I assume you don't have an SDNode* lvalue to return a reference to) I assume the adapter helper can implement one of those in terms of the other so you only have to implement one of them? I forget how the adapter utility works.
> I think it makes sense to do this. This will unfortunately be one of the few SDNode & in the entire codebase though, which makes it stand out. SDNode really does seem to always be a pointer. I’ll fix up the patch to do this soon.
>
> Yeah, there are a few types (read: Lots) like this in LLVM. I personally don't mind being more referential in spite of that, but I can understand others might feel less comfortable with that.
>
> If you wanted to preserve the pointer-ness, you'd have op* return SDNode * (this would be a bit incorrect, it should really be SDNode * const &, and you can do that by having an SDNode* member in the iterator that you init and return a ref to... technically that's the more correct option - I'm not entirely sure where that matters) and no op->.
I like this solution. I tried SDNode * const & earlier but of course getNode() is a temporary so this doesn’t work. I’ll get a patch together which makes this change. Technically I guess that means we don’t need the nonnull SDNode patch, but I don’t see any harm in it anyway.
Thanks,
Pete
>>
>>
>> BTW, been trying to work out if there would ever be a good solution for an iterator combined with isa<> or dyn_cast<>. If you look at the code this patch touches in AArch64ISelLowering, it is immediately followed by a dyn_cast. I’d really like to find a clean way to fold that it to the foreach loop, i.e.,
>>
>> for (auto *C : dyn_cast<ConstantSDNode>(N->op_nodes()))
>>
>> just a thought, but thats unrelated to this patch for now.
>>
>> Yep, though probably more in the form of a filtered range, I suspect:
>>
>> for (auto &C : filtered_transform(N->op_nodes(), [](SDUse *U) { return U->getNode(); }))
>>
>> It'd be a bit tricky to deal with the value type of this range's iterators - chances are the predicate should return an Optional<T&> (Hmm, don't think our Optional template supports ref parameters yet anyway) or T* (not sure if we could generalize it so it could cope with Optional<T>, maybe - so we could support generators where the values are not already/permanently in-memory) and then the value_type is T.
> Interesting. I hadn’t though to use Optional. I might try to implement something like this if i get time.
>
> Cheers,
> Pete
>>
>>
>> Cheers,
>> Pete
>>
>>
>
>
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20150625/186d7c6a/attachment.html>
More information about the llvm-commits
mailing list