summaryrefslogtreecommitdiff
path: root/tools/jit/gen_jit_dispatch.py
AgeCommit message (Expand)AuthorFilesLines
2019-04-02Adding pin_memory kwarg to zeros, ones, empty, ... tensor constructors. (#18455)Vitaly Fedyunin1-14/+19
2019-03-30Turn on F401: Unused import warning. (#18598)Edward Yang1-5/+2
2019-03-15Fix truncation of default float values in JIT signatures. (#18044)Gregory Chanan1-1/+5
2019-03-12Remove (almost all) TensorOptions from native_functions.yaml (#17385)Christian Puhrsch1-5/+23
2019-03-07Use return names in JIT operatorsChristian Puhrsch1-0/+3
2019-02-27Use name for output variables instead of out in JIT (#17386)Christian Puhrsch1-1/+1
2019-02-10Support named tuple return from operators on JIT (#16253)Xiang Gao1-1/+3
2019-02-07Remove undefined tensor in jit script (#16379)Wanchao Liang1-17/+48
2019-02-05Rename IntList to IntArrayRef. (#16751)Edward Yang1-5/+5
2019-01-17Add matches_jit_signature attribute to native_functions.yaml (#16040)Christian Puhrsch1-1/+12
2019-01-03Torch tensor (#15224)Elias Ellison1-3/+3
2018-12-21Remove NoneGeneratorDavid Riazati1-1/+1
2018-12-19Remove python_default_init from ATen and use Optional (#15234)Wanchao Liang1-0/+2
2018-12-19Optional ScalarType support for native functions & JIT (#15154)Tugrul Ates1-0/+2
2018-12-03Implement Device as a type in the script (#14666)Zachary DeVito1-22/+4
2018-11-09Optional undefined tensor support (#13650)Wanchao Liang1-0/+3
2018-11-06Add README to tools, delete defunct scripts. (#13621)Edward Yang1-0/+14
2018-11-01Rename elementwise_mean to mean (#13419)Tongzhou Wang1-1/+1
2018-10-29Add tensor.to(options) (#13146)Tongzhou Wang1-36/+54
2018-10-26Bind inplace and _out variants into JIT (#13093)Zachary DeVito1-26/+97
2018-10-25Add c10::optional to type syntax (#12582)Wanchao Liang1-4/+3
2018-10-15split register_aten_ops.cpp into shards (#12615)Anders Papitto1-15/+26
2018-10-03Add `bool` type to IR (#11834)David Riazati1-1/+1
2018-09-24Make gen_jit_dispatch runnable (#12018)Bram Wasti1-1/+1
2018-09-11Make .to() methods native functions (to fix JIT tracing)Adam Paszke1-1/+1
2018-09-11Allow tracing random functions (only when using default generators) (#11539)Adam Paszke1-6/+4
2018-09-11Add support for tracing strings (#11506)Adam Paszke1-1/+3
2018-09-11Improve shape analysis to cover all most commonly used ops (#11358)Adam Paszke1-3/+3
2018-08-29Record tensor factory functions in trace (#10935)Zachary DeVito1-1/+2
2018-08-02Remove capture specifiers in register_aten_ops when they're not needed. (#9669)Owen Anderson1-1/+3
2018-08-01Simplified Operator (#10080)Peter Goldsborough1-65/+8
2018-07-31Add (constant size) TensorLists to JIT, use them in cat and stack nodes (#9948)Adam Paszke1-59/+14
2018-07-26Unify IR operator representation (stop using attributes in the JIT) (#9807)Adam Paszke1-1/+1
2018-07-25added torch.rot90() to ATen (#8628)Wei Yang1-1/+3
2018-07-23Switch interpreter to use IValue's primitive int/floats (#9718)Zachary DeVito1-77/+65
2018-07-20Prepare to stop using attributes in the JIT (#9505)Adam Paszke1-0/+1
2018-07-16Add a tagged union type that replaces tensor in the interpreter. (#9368)Zachary DeVito1-4/+5
2018-07-10Unify aten_dispatch and aten_schema into a single operator abstraction with h...Zachary DeVito1-166/+168
2018-06-22Cache cufft plans (#8344)Tongzhou Wang1-0/+4
2018-06-21Better support for literals in jit script (#8687)Richard Zou1-1/+36
2018-06-16Create ATen tensors via TensorOptions (#7869)Peter Goldsborough1-22/+25
2018-06-15Port THS to ATen. (#8409)Edward Z. Yang1-1/+3
2018-06-12changes to support ATen code generation inside fbcode (#8397)anderspapitto1-10/+10
2018-06-07Rename SparseTensor to SparseTensorRef. (#8237)Edward Z. Yang1-1/+1
2018-06-06Add more annotations for arguments in ATen schema (#8192)Adam Paszke1-2/+3
2018-06-05[JIT] Support a single TensorList argument anywhere in the argument list + in...James Reed1-11/+39
2018-06-01[script] Add support for torch.zeros, torch.ones, etc. (#7799)Zachary DeVito1-24/+67
2018-05-17Reduce gen_jit_dispatch options (#7562)Zachary DeVito1-18/+10
2018-05-14Improve script builtin checking using schema (#7311)Zachary DeVito1-17/+92
2018-04-29[jit] Fix handling of IntList[k] parameters (#6965)Marcin Elantkowski1-10/+41