saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
24355 commits 5021 branches 1206 tags 1 GiB
Commit graph

4 commits

Author SHA1 Message Date
chronos_secgrp_pytorch_oss_ci_oncall
e71a0f5247 2022-03-01 nightly release (5e86505693) 2022-03-01 00:02:25 -08:00
chronos_secgrp_pytorch_oss_ci_oncall
195f29e3ba 2020-06-11 nightly release (c3d4053bc0) 2020-06-11 00:02:27 -07:00
Peter Goldsborough
b70b7a80d4 Inline JIT C++ Extensions (#7059) 
Adds ability to JIT compile C++ extensions from strings

>>> from torch.utils.cpp_extension import load_inline
>>> source = '''
    at::Tensor sin_add(at::Tensor x, at::Tensor y) {
      return x.sin() + y.sin();
    }
'''
>>> module = load_inline(name='inline_extension', cpp_sources=source, functions='sin_add')
Fixes #7012

* Inline JIT C++ Extensions

* jit_compile_sources -> jit_compile

* Split up test into CUDA and non-CUDA parts

* Documentation fixes

* Implement prologue and epilogue generation

* Remove extra newline

* Only create the CUDA source file when cuda_sources is passed
 2018-04-30 11:48:44 -04:00
Peter Goldsborough
792daeb422 Enable documentation for C++ extensions on the website (#5597) 2018-03-07 14:07:26 +01:00