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 37 Packages Projects Releases Wiki Activity
ee21142e40
pytorch/caffe2/core
History
Ilia Cherniavskii 7190789f58 Handling of failing and terminal async cpu ops (#29052) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29052

Make sure we handle the case of multiple, async, terminal (no children)
and failing cpu ops.

Test Plan: AsyncIf tests

Reviewed By: yyetim

Differential Revision: D18276401

Pulled By: ilia-cher

fbshipit-source-id: 35b175dd025bc7e392056ac1331b159376a29e60
2019-11-04 12:01:21 -08:00
..
hip Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
nomnigraph cleanup warnings 2019-08-12 16:12:30 -07:00
__init__.py re-enable copy of python files, but be careful that the copy is only … (#14982) 2018-12-11 16:54:08 -08:00
allocator.cc unify c2 and TH allocator (#16892) 2019-02-12 21:16:34 -08:00
allocator.h unify c2 and TH allocator (#16892) 2019-02-12 21:16:34 -08:00
asan.h
blob_gpu_test.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
blob_serialization_gpu.cc
blob_serialization.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
blob_serialization.h fix -Wsign-compare warnings for some files inside c2 (#18123) 2019-03-19 10:39:20 -07:00
blob_serializer_base.h Serialization takes pointers instead of Blob (#11925) 2018-10-17 11:50:34 -07:00
blob_stats.cc
blob_stats.h Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
blob_test.cc Apply modernize-use-override (2nd iteration) 2019-02-14 16:52:57 -08:00
blob.h Adding quantized tensor shape/type info support for caffe2=>glow in caffe2 side (#18621) 2019-03-31 17:42:27 -07:00
CMakeLists.txt get rid of protobuf dependencies (#25650) 2019-09-06 08:48:20 -07:00
common_cudnn.cc
common_cudnn.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
common_gpu.cc switch to rocThrust for thrust/cub APIs (#25620) 2019-09-03 22:16:30 -07:00
common_gpu.h Torch rename (#20774) 2019-06-12 20:12:34 -07:00
common_omp.h
common_test.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
common.cc
common.h Torch rename (#20774) 2019-06-12 20:12:34 -07:00
context_base.cc Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
context_base.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
context_gpu_test.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
context_gpu.cu Lightweight at-most-once logging for API usage (#20745) 2019-05-23 23:17:59 -07:00
context_gpu.h Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
context_test.cc unify c2 and TH allocator (#16892) 2019-02-12 21:16:34 -08:00
context.cc Change Tensor::CopyFrom to a simple double dispatch (#14268) 2018-11-28 15:45:37 -08:00
context.h Fix a lot of C++ build warnings (#16411) 2019-01-31 14:35:56 -08:00
cudnn_wrappers.h Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
db.cc Apply modernize-use-override (2nd iteration) 2019-02-14 16:52:57 -08:00
db.h Change message on unknown db type to be friendly (#17795) 2019-03-08 10:46:24 -08:00
event_cpu.h Delete unnecessary include from allocator.cc/event_cpu.h 2018-09-19 16:45:54 -07:00
event_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
event_gpu.cc Replace caffe2::DeviceGuard with c10::cuda::CUDAGuard (#17623) 2019-03-06 10:48:15 -08:00
event_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
event.cc
event.h Fix issues under caffe round 1 2019-01-23 19:04:59 -08:00
export_c10_op_to_caffe2.cc Rename caffe2<->c10 operator wrappers (#21322) 2019-06-07 13:48:10 -07:00
export_c10_op_to_caffe2.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
export_caffe2_op_to_c10.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
flags.h move flags to c10 (#12144) 2018-10-04 02:09:56 -07:00
graph_test.cc
graph.cc
graph.h
init_denormals.cc add denormal options (ftz and daz) 2018-12-20 17:04:39 -08:00
init_intrinsics_check.cc cleanup warnings 2019-08-12 16:12:30 -07:00
init_omp.cc Using c10 namespace across caffe2. (#12714) 2018-10-17 12:57:19 -07:00
init_test.cc Allow for registration after GlobalInit (#15876) 2019-01-10 09:35:33 -08:00
init.cc Update intra_inter_benchmark (#22051) 2019-06-21 23:06:27 -07:00
init.h Update intra_inter_benchmark (#22051) 2019-06-21 23:06:27 -07:00
int8_serialization.cc Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
logging.h Moving logging from caffe2 to c10. (#12881) 2018-10-19 20:22:08 -07:00
macros.h
macros.h.in Revert D16929363: Revert D16048264: Add static dispatch mode to reduce mobile code size 2019-08-20 17:08:31 -07:00
memonger.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
memonger.h
module_test.cc
module.cc cleanup warnings 2019-08-12 16:12:30 -07:00
module.h cleanup warnings 2019-08-12 16:12:30 -07:00
net_async_base.cc Handling of failing and terminal async cpu ops (#29052) 2019-11-04 12:01:21 -08:00
net_async_base.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
net_async_scheduling.cc add cancelAsyncCallback method to OperatorBase (#21492) 2019-06-06 20:57:12 -07:00
net_async_scheduling.h Reinforce scheduling invariants (#17132) 2019-02-20 16:32:23 -08:00
net_async_task_future.cc cleanup warnings 2019-08-12 16:12:30 -07:00
net_async_task_future.h Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task_graph.cc Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task_graph.h Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task.cc Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_task.h Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor" 2018-12-10 19:30:58 -08:00
net_async_tracing_test.cc AsyncNet: option for time based tracing and trace path (#13440) 2018-11-08 11:34:34 -08:00
net_async_tracing.cc Add batch id to tracer event (#21446) 2019-06-13 17:13:42 -07:00
net_async_tracing.h Add batch id to tracer event (#21446) 2019-06-13 17:13:42 -07:00
net_dag_utils_test.cc New chaining/partitioning algorithm for async_scheduling for inference (#11957) 2018-10-08 12:24:52 -07:00
net_dag_utils.cc Back out "Back out "[Caffe2] Fix device_option propagation"" (#25908) 2019-09-17 04:01:36 -07:00
net_dag_utils.h Back out "[nomnigraph][executor] computeChains with nomnigraph" (#15451) 2018-12-21 11:09:27 -08:00
net_gpu_test.cc Remove unused executors, part 3 (#14199) 2018-11-26 19:10:43 -08:00
net_parallel.cc Open registration for c10 thread pool (#17788) 2019-03-08 15:38:41 -08:00
net_parallel.h TBB task graph (#15041) 2018-12-10 21:35:04 -08:00
net_simple_refcount_test.cc Remove unused executors, part 3 (#14199) 2018-11-26 19:10:43 -08:00
net_simple_refcount.cc net_simple_refcount type to help experimentation with dynamic allocation. (#13370) 2018-10-31 15:59:16 -07:00
net_simple_refcount.h net_simple_refcount type to help experimentation with dynamic allocation. (#13370) 2018-10-31 15:59:16 -07:00
net_simple.cc Back out "Back out "[Caffe2] Fix device_option propagation"" (#25908) 2019-09-17 04:01:36 -07:00
net_simple.h Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
net_test.cc Handling of failing and terminal async cpu ops (#29052) 2019-11-04 12:01:21 -08:00
net.cc Lightweight at-most-once logging for API usage (#20745) 2019-05-23 23:17:59 -07:00
net.h Removing cyclic dependency (#20511) 2019-05-14 18:55:19 -07:00
numa.cc Move numa.{h, cc} to c10/util (#15024) 2018-12-12 12:21:10 -08:00
numa.h Move numa.{h, cc} to c10/util (#15024) 2018-12-12 12:21:10 -08:00
observer_test.cc Apply modernize-use-override (2nd iteration) 2019-02-14 16:52:57 -08:00
observer.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
operator_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
operator_gradient.h Fix "missing return statement" warning (#22216) 2019-06-25 16:57:42 -07:00
operator_schema_test.cc
operator_schema.cc Add filler for SparseLengthsWeightedSum (#13949) 2018-11-16 11:31:05 -08:00
operator_schema.h Verify def before infer fensor (#18129) 2019-03-22 06:36:25 -07:00
operator_test.cc Register gradient op with engine (#21205) 2019-05-31 18:48:47 -07:00
operator.cc Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
operator.h Support Offline Tensors through ONNXIFI layer 2019-10-31 10:33:42 -07:00
parallel_net_test.cc
plan_executor.cc Fix spelling errors (#21665) 2019-06-13 15:21:55 -07:00
plan_executor.h
prof_dag_counters.cc Added engine information to the profiling result. (#20493) 2019-05-28 16:41:12 -07:00
prof_dag_counters.h Better handling of net errors in prof_dag counters (#17384) 2019-02-22 18:38:31 -08:00
qtensor_serialization.cc
qtensor_serialization.h Guard all Caffe2 protobuf string serializations with CAFFE_ENFORCE (fixed reverted bug) (#12848) 2018-10-23 16:21:26 -07:00
qtensor.cc
qtensor.h rest of uses for deprecation of dims() in Tensor (#16118) 2019-01-18 11:52:12 -08:00
scope_guard.h
static_tracepoint_elfx86.h
static_tracepoint.h
stats_test.cc
stats.cc
stats.h Add full namespace resolution in CAFFE_DURATION (#12065) 2018-09-26 13:29:18 -07:00
storage.h Move files to/from c10/core and c10/util (#15316) 2019-01-10 16:22:22 -08:00
tensor_impl.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
tensor_int8.cc Fix include paths for typeid.h (#13689) 2018-11-14 18:04:09 -08:00
tensor_int8.h
tensor.cc Back out "Revert D17936166: [wip] Constexpr type ids" (#28155) 2019-10-16 18:24:04 -07:00
tensor.h Remove preallocation of type ids (#28024) 2019-10-16 23:08:11 -07:00
test_utils.cc Fuse to individual operators to GatherFuse8BitRowwiseQuantFloatMulLengthElim (#25519) 2019-09-03 19:08:49 -07:00
test_utils.h Add randomFill to test_utils.h 2019-10-04 18:29:22 -07:00
timer_test.cc
timer.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
transform_test.cc
transform.cc Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
transform.h Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
types.cc Add support to serialize qtensor in JIT. (#23356) 2019-07-26 15:52:15 -07:00
types.h move GetDimFromOrderString to caffe2/core/types.h (#25671) 2019-09-05 04:32:04 -07:00
workspace_test.cc
workspace.cc move flags to c10 (#12144) 2018-10-04 02:09:56 -07:00
workspace.h Fix workspace destruction ordering (#23096) 2019-07-19 16:49:50 -07:00