OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
f51a497f92
pytorch/torch/csrc/jit/frontend
History
David Berard f51a497f92 [JIT] add builtin functions for (complex, Tensor) (#73286) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73286

Adds support in jit for op(complex, Tensor) for the following:
* add (+)
* mul (*)
* eq (==)
* ne (!=)
* sub (-)
* div (/)

These already worked in eager mode, but didn't work in jit.

Test Plan: Imported from OSS

Reviewed By: george-qi

Differential Revision: D34421794

Pulled By: davidberard98

fbshipit-source-id: 0aecec7d7da1dcf2bcceb6849d3049fe2b39076d
(cherry picked from commit 375998d38d60ab97f7be8d95dd7bdc10bd1f6417)
2022-02-24 18:40:47 +00:00
..
builtin_functions.cpp [JIT] add builtin functions for (complex, Tensor) (#73286) 2022-02-24 18:40:47 +00:00
builtin_functions.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
canonicalize_modified_loop.cpp Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
canonicalize_modified_loop.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
concrete_module_type.cpp
concrete_module_type.h
convert_to_ssa.cpp
convert_to_ssa.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
edit_distance.cpp
edit_distance.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
error_report.cpp
error_report.h
exit_transforms.cpp
exit_transforms.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
function_schema_parser.cpp Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
function_schema_parser.h
inline_loop_condition.cpp Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
inline_loop_condition.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
ir_emitter.cpp (2/2) Make TorchScript Preserve Fully Qualified Class Name for Python Exceptions: frontend change (#72899) 2022-02-16 03:45:15 +00:00
ir_emitter.h
lexer.cpp
lexer.h Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
mini_environment.h
name_mangler.cpp
name_mangler.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
parse_string_literal.h
parser_constants.h
parser.cpp Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
parser.h Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
resolver.h
schema_matching.cpp [jit][edge] Migrate to TypeFactory for jit types on mobile (#71516) 2022-01-26 07:32:04 +00:00
schema_matching.h Bump version number to 7 and compile old operators with old schema (#68358) 2022-01-05 23:57:22 -08:00
schema_type_parser.cpp [jit][edge] Migrate to TypeFactory for jit types on mobile (#71516) 2022-01-26 07:32:04 +00:00
schema_type_parser.h [jit] Reclaim some binary size. (#68038) 2022-01-07 11:21:46 -08:00
script_type_parser.cpp Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
script_type_parser.h Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00
source_range.cpp Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
source_range.h Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
source_ref.h Revert D33994011: Make debug_pkl smaller by only emitting unique traces. 2022-02-24 16:38:55 +00:00
strtod.cpp
strtod.h
sugared_value.cpp
sugared_value.h (2/2) Make TorchScript Preserve Fully Qualified Class Name for Python Exceptions: frontend change (#72899) 2022-02-16 03:45:15 +00:00
tracer.cpp [AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT 2022-01-02 17:30:39 -08:00
tracer.h [AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT 2022-01-02 17:30:39 -08:00
tree_views.cpp caffe2/torch/csrc/jit/frontend/tree_views: workaround nvcc compiler error 2022-01-14 21:58:31 -08:00
tree_views.h [caffe2][torch] Remove unreferenced local variable e (#71856) 2022-01-28 21:00:37 +00:00
tree.h Reland D33284352: [jit][edge] Do not reuse mobile type parser for all unpicklers. (#71048) 2022-01-10 12:42:23 -08:00
versioned_symbols.cpp Fix inconsistency between new and old upgrader design (#71185) 2022-01-12 09:54:31 -08:00
versioned_symbols.h Fix inconsistency between new and old upgrader design (#71185) 2022-01-12 09:54:31 -08:00