[swift-evolution] Compile-time generic specialization
Abe Schneider
abe.schneider at gmail.com
Sun Feb 5 11:37:15 CST 2017
Hi Robert,
Sorry, I’m not sure I understand your question. In c++ you can do the following:
struct Storage {};
struct CBlasStorage: Storage {};
template <typename S> class Tensor {};
template <typename S>
Tensor<S> dot(const Tensor<S> &lhs, const Tensor<S> &rhs) {
std::cout << "general version called" << std::endl;
Tensor<S> result;
return result;
}
// specialized version for CBlasStorage
template <>
Tensor<CBlasStorage> dot(const Tensor<CBlasStorage> &lhs, const Tensor<CBlasStorage> &rhs) {
std::cout << "specialized version called" << std::endl;
Tensor<CBlasStorage> result;
return result;
}
// this preserves type information and will call the appropriate `dot`
template <typename T>
void doSomething(const Tensor<T> &lhs, const Tensor<T> &rhs) {
auto result = dot(lhs, rhs);
}
int main(int argc, char **argv) {
Tensor<CBlasStorage> a, b;
doSomething(a, b); // we should get "specialized version called"
}
The potential equivalent for Swift could look like:
@_specialize_all
func dot<S:Storage>(_ lhs:Tensor<S>, _ rhs:Tensor<S>) -> Tensor<S> { … }
Which would cause the compile to create a version of `dot` per S type that it gets called with. Thus, when `doSomething` is called, it would dispatch to that version of `dot`, allowing the type information to be preserved in the same way it does in c++.
Abe
> On Feb 5, 2017, at 11:35 AM, Robert Widmann <devteam.codafi at gmail.com> wrote:
>
> I don't understand how this change would cause method dispatch to invoke a different prototype. Specialization in either language mentioned doesn't do that.
>
> ~Robert Widmann
>
> 2017/02/05 11:28、Abe Schneider via swift-evolution <swift-evolution at swift.org> のメッセージ:
>
>> Hi all,
>>
>> The current behavior of generics in Swift causes it lose type information at compile time due to the desire of maintaining a single version of the function. This runs counter to how c++ works, which creates a new copy of a function per type, but preserves information to be preserved. This can cause unexpected behavior from the user’s perspective:
>>
>> protocol DispatchType {}
>> class DispatchType1: DispatchType {}
>>
>> func doBar<D:DispatchType>(value:D) {
>> print(“General function called")
>> }
>>
>> func doBar(value:DispatchType1) {
>> print("DispatchType1 called")
>> }
>>
>> func test<D:DispatchType>(value:D) {
>> doBar(value: value)
>> }
>>
>> test(value: d1) // “General function called”, but it’s not obvious why
>>
>>
>> The suggested method to get around this issue is to use a protocol to create a witness table, allowing for runtime dispatch. However, this approach is not ideal in all cases because: (a) the overhead of runtime dispatch may not be desirable, especially because this is something that can be determined at compile time; and (b) there are some designs in which this behavior can complicate things.
>>
>> One example of a design where this behavior can be problematic is when a protocol is used to determine what functions get dispatched:
>>
>> protocol Storage { … }
>> class Tensor<S:Storage> { … }
>>
>> class CBlasStorage: Storage { … }
>> class OpenCLStorage: Storage { … }
>>
>> func dot<S:Storage>(_ lhs:Tensor<S>, _ rhs:Tensor<S>) -> Tensor<S> { … }
>>
>> // like behavior, these will not work if called from another generic function (but will work for non-generic functions)
>> func dot<S:Storage>(_ lhs:Tensor<S>, _ rhs:Tensor<S>) -> Tensor<S> where S:CBlasStorage { … }
>> func dot<S:Storage>(_ lhs:Tensor<S>, _ rhs:Tensor<S>) -> Tensor<S> where S:OpenCLStorage { … }
>>
>> In this case, depending on the underlying storage, we want an optimized version of `dot` to be called. To make this work correctly we can add static methods to `Tensor`, but this has several drawbacks: (a) it makes the `Tensor` class monolithic, every possible method must be determine a priori and be defined in the class; (b) it doesn’t allow new methods to be added Tensor without touching the main class; and (c) it unnecessarily forces users to user the more verbose `Tensor.dot(a, b)`.
>>
>> Point (a) in theory could be made better by creating a `TensorOps` protocols. However, because type constraints cannot currently be placed on extensions, it is not currently possible to implement.
>>
>>
>> One potential solution would be to add/extend an attribute for generic functions that would force multiple versions of that function to be created. There is already there is a `@_specialize` attribute, but you have to: (a) manually write out all the cases you want to cover; and (b) only affects the compiled code, which does not change this behavior. Due to the fact that `@_specialize` exists, I’m going to assume it wouldn’t be a major change to the language to extend the behavior to compile-time dispatch.
>>
>>
>> Thanks!
>> Abe
>> _______________________________________________
>> swift-evolution mailing list
>> swift-evolution at swift.org
>> https://lists.swift.org/mailman/listinfo/swift-evolution
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://lists.swift.org/pipermail/swift-evolution/attachments/20170205/fd6cbb3c/attachment.html>
More information about the swift-evolution
mailing list