pytorch/torch/csrc/jit/codegen/cuda
Sebastian Messmer 53af9df557 Unify boxed function signature between jit and c10 (#37034)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37034

c10 takes a Stack* in boxed functions while JIT took Stack&.
c10 doesn't return anything while JIT returns an int which is always zero.

This changes JIT to follow the c10 behavior.
ghstack-source-id: 106834069

Test Plan: unit tests

Differential Revision: D20567950

fbshipit-source-id: 1a7aea291023afc52ae706957e9a5ca576fbb53b
2020-06-29 19:24:26 -07:00
..
arith.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
arith.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
dispatch.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
dispatch.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
expr_evaluator.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
expr_evaluator.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
fusion.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
fusion.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
graph_fuser.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
index_compute.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
index_compute.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
interface.cpp Unify boxed function signature between jit and c10 (#37034) 2020-06-29 19:24:26 -07:00
interface.h
ir_all_nodes.h New CUDA Fuser code lowering refactor (#36199) 2020-04-09 14:27:05 -07:00
ir_base_nodes.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_base_nodes.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_graphviz.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_graphviz.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_interface_nodes.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_internal_nodes.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_iostream.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_iostream.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_nodes.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
ir_printer.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
iter_visitor.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
iter_visitor.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
kernel_arg.h [CUDA_FUSER] Expand operation support for cuda fuser (#37849) 2020-05-07 09:21:09 -07:00
kernel_cache.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
kernel_cache.h [resubmit] Use TensorMethods.cpp (#39385) 2020-06-02 20:27:51 -07:00
kernel_resource_strings.h [nvFuser] Reduction support in codegen, fp16 support (#38627) 2020-05-21 17:18:39 -07:00
kernel.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
kernel.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
lower_loops.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
lower_loops.h [nvFuser] Reduction support in codegen, fp16 support (#38627) 2020-05-21 17:18:39 -07:00
lower_utils.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
lower_utils.h [nvFuser] Reduction support in codegen, fp16 support (#38627) 2020-05-21 17:18:39 -07:00
lower2device.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
lower2device.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
manager.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
manager.h
mutator.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
mutator.h New CUDA Fuser: Unrolling support, interface refactor (#36435) 2020-04-16 09:20:24 -07:00
parser.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
parser.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
partition.cpp [CUDA_FUSER] Expand operation support for cuda fuser (#37849) 2020-05-07 09:21:09 -07:00
partition.h
predicate_compute.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
predicate_compute.h [nvFuser] Reduction support in codegen, fp16 support (#38627) 2020-05-21 17:18:39 -07:00
register_interface.cpp
shape_inference.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
shape_inference.h [CUDA_FUSER] Expand operation support for cuda fuser (#37849) 2020-05-07 09:21:09 -07:00
tensor_meta.cpp [nvFuser] Reduction support in codegen, fp16 support (#38627) 2020-05-21 17:18:39 -07:00
tensor_meta.h
tensor_view.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
transform_iter.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
transform_iter.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
transform_replay.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
transform_replay.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
transform_rfactor.cpp [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
transform_rfactor.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
type.cpp Get rid of global constructors in cuda codegen (#40183) 2020-06-18 13:54:11 -07:00
type.h [nvFuser] Transform replay refactor and minor updates (#39579) 2020-06-11 23:04:24 -07:00
utils.cpp Adding symbolic sizes, contiguity, stride indices (#36101) 2020-05-01 02:01:25 -07:00
utils.h Adding symbolic sizes, contiguity, stride indices (#36101) 2020-05-01 02:01:25 -07:00