Pruthvi Madugundu fbd08fb358 Introduce TORCH_DISABLE_GPU_ASSERTS (#84190) ... - Asserts for CUDA are enabled by default - Disabled for ROCm by default by setting `TORCH_DISABLE_GPU_ASSERTS` to `ON` - Can be enabled for ROCm by setting above variable to`OFF` during build or can be forcefully enabled by setting `ROCM_FORCE_ENABLE_GPU_ASSERTS:BOOL=ON` This is follow up changes as per comment in PR #81790, comment [link](https://github.com/pytorch/pytorch/pull/81790#issuecomment-1215929021) Pull Request resolved: https://github.com/pytorch/pytorch/pull/84190 Approved by: https://github.com/jeffdaily, https://github.com/malfet		2022-11-04 04:43:05 +00:00
..
hip
nomnigraph	Update CMakeLists.txt (#87030)	2022-10-28 04:56:40 +00:00
__init__.py
allocator.cc
allocator.h
blob.h
blob_gpu_test.cc
blob_serialization.cc
blob_serialization.h
blob_serialization_gpu.cc
blob_serializer_base.h
blob_stats.cc
blob_stats.h
blob_test.cc
CMakeLists.txt	Remove caffe2 mobile (#84338)	2022-09-08 01:49:55 +00:00
common.cc
common.h
common_cudnn.cc
common_cudnn.h
common_gpu.cc
common_gpu.h
common_omp.h
common_test.cc
context.cc
context.h
context_base.cc
context_base.h
context_gpu.cu	Check all CUDA API calls for errors in caffe2/ (#81816)	2022-10-28 00:41:06 +00:00
context_gpu.h	Check all CUDA API calls for errors in caffe2/ (#81816)	2022-10-28 00:41:06 +00:00
context_gpu_test.cc
context_test.cc
cudnn_wrappers.h
db.cc
db.h
distributions_stubs.h
event.cc
event.h
event_cpu.h
event_gpu.cc
event_gpu_test.cc
event_test.cc
export_c10_op_to_caffe2.cc
export_c10_op_to_caffe2.h
export_caffe2_op_to_c10.h
flags.h
graph.cc
graph.h
graph_test.cc
init.cc
init.h
init_denormals.cc
init_intrinsics_check.cc
init_omp.cc
init_test.cc
int8_serialization.cc
logging.h
macros.h
macros.h.in	Introduce TORCH_DISABLE_GPU_ASSERTS (#84190)	2022-11-04 04:43:05 +00:00
memonger.cc
memonger.h
module.cc
module.h
module_test.cc
net.cc
net.h
net_async_base.cc
net_async_base.h
net_async_scheduling.cc
net_async_scheduling.h
net_async_task.cc
net_async_task.h
net_async_task_future.cc
net_async_task_future.h
net_async_task_graph.cc
net_async_task_graph.h
net_async_tracing.cc
net_async_tracing.h
net_async_tracing_test.cc
net_dag_utils.cc
net_dag_utils.h
net_dag_utils_test.cc
net_gpu_test.cc
net_parallel.cc
net_parallel.h
net_simple.cc
net_simple.h
net_simple_refcount.cc
net_simple_refcount.h
net_simple_refcount_test.cc
net_test.cc
numa.cc
numa.h
observer.h
observer_test.cc
operator.cc
operator.h
operator_gpu_test.cc
operator_gradient.h
operator_schema.cc
operator_schema.h	Revert "Fix crash on unload torch cpu dll (#67632)"	2022-08-02 00:56:18 +00:00
operator_schema_test.cc
operator_test.cc
parallel_net_test.cc
plan_executor.cc
plan_executor.h
plan_executor_test.cc
prof_dag_counters.cc
prof_dag_counters.h
qtensor.cc
qtensor.h
qtensor_serialization.cc
qtensor_serialization.h
scope_guard.h
serialization_test.cc
static_tracepoint.h
static_tracepoint_elfx86.h
stats.cc
stats.h
stats_test.cc
storage.h
tensor.cc	[PyTorch][caffe2] Add CAFFE2_{DECLARE,DEFINE}_KNOWN_TYPE (#83707)	2022-08-30 23:09:49 +00:00
tensor.h	[PyTorch][caffe2] Add CAFFE2_{DECLARE,DEFINE}_KNOWN_TYPE (#83707)	2022-08-30 23:09:49 +00:00
tensor_impl.h
tensor_int8.cc
tensor_int8.h
test_utils.cc
test_utils.h
timer.h
timer_test.cc
transform.cc
transform.h
transform_test.cc
types.cc
types.h
workspace.cc
workspace.h
workspace_test.cc