perf: Move task_pt_regs sampling into arch code