const ProfilerVoidEventStub* event2_) const override { auto event1 = event1_->get(); auto event2 = event2_->get(); TORCH_NPU_CHECK(aclrtSynchronizeEvent(event1)); ASCEND_LOGI("Event: aclrtSynchronizeEvent is successfully executed for event1."); TORCH_NPU_CHECK(aclrtSynchronize...
self._action_controller=ActionController(self._msprofiler_interface,schedule,self,on_trace_ready) self._use_cuda=use_cuda self._check_params() def__enter__(self): _lazy_init() self._action_controller.transit_action() returnself def__exit__(self,exe_type,exe_val,exc_tb): ...
"libascend_ml.so"] other_dependencies = ["libtorch_npu.so", "libnpu_profiler.so", "libgomp.so.1", "libatb.so"] dependencies = torch_dependencies + cann_dependencies + other_dependencies bdist_wheel.run(self) if is_manylinux: file = glob.glob(os.path.join(self.dist_dir, "*...
torch_npu._C._enable_e2e_profiler(self.result_path,self.npu_event|npuEvent().ACL_PROF_MSPROFTX, self.aicore_metrics) else: prof_init(self.result_path) prof_start(self.npu_event,self.aicore_metrics) returnself def__exit__(self,exc_type,exc_val,exc_tb): ...
torch_npu._C._profiler._prepare_profiler(self.config(), self.kineto_activities) def __exit__(self, exc_type, exc_val, exc_tb): if not self.enabled: return records = torch_npu._C._profiler._disable_profiler_legacy() parsed_results = parse_legacy_records(records) self.function_...