Searched refs:assertions_data (Results 1 – 2 of 2) sorted by relevance
21 DeviceAssertionsData* assertions_data, in dsa_add_new_assertion_failure() argument32 if (!assertions_data) { in dsa_add_new_assertion_failure()40 const auto nid = atomicAdd(&(assertions_data->assertion_count), 1); in dsa_add_new_assertion_failure()54 auto& self = assertions_data->assertions[nid]; in dsa_add_new_assertion_failure()78 assertions_data, \
159 [[maybe_unused]] c10::cuda::DeviceAssertionsData *const assertions_data, \164 #define TORCH_DSA_KERNEL_ARGS_PASS assertions_data, assertion_caller_id