projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 24752eb) | patch
Introduce lambda-based kernel API (#18541)
authorSebastian Messmer <messmer@fb.com>
Sat, 30 Mar 2019 07:03:46 +0000 (00:03 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 30 Mar 2019 07:07:31 +0000 (00:07 -0700)
commitf4e87e193a06309923ed1b5df153d4a1922459c8
tree641da434822209ab66583224e550fc15f7905dcdtree | snapshot
parent24752eb7b827607ad0ff65e89d6c25b5bf48896acommit | diff
Introduce lambda-based kernel API (#18541)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18541

Allow registering lambdas as c10 kernels.

Reviewed By: dzhulgakov

Differential Revision: D14653005

fbshipit-source-id: f867cc776b1339e83b7a2e1935f5cf924cfba44a
aten/src/ATen/core/op_registration/kernel_functor.h diff | blob | history
aten/src/ATen/core/op_registration/kernel_lambda.h [new file with mode: 0644] blob
aten/src/ATen/core/op_registration/kernel_lambda_test.cpp [new file with mode: 0644] blob
aten/src/ATen/core/op_registration/op_registration.h diff | blob | history
c10/test/util/TypeTraits_test.cpp diff | blob | history
c10/util/Metaprogramming.h diff | blob | history
c10/util/TypeTraits.h diff | blob | history
Domain: Machine Learning / ML Framework;