OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix ios_crash:backtrace=FBCameraFramework:caffe2::getClockTimeMilliseconds() (perf_observer.cc (#24813)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24813

clock_gettime does not work on some apple platforms e.g. OSX < 10.12

Use a custom implementation, similar to https://opensource.apple.com/source/Libc/Libc-1158.1.2/gen/clock_gettime.c.auto.html

T52655182

Test Plan: sandcastle tests

Differential Revision: D16883407

fbshipit-source-id: a42828bb91bb0c43297e9bdce4b18f7c9ea4274d
This commit is contained in:
Geoffrey Goh 2019-08-24 21:14:34 -07:00 committed by Facebook Github Bot
parent f2bcad5ddf
commit b7b80c6bdd
1 changed files with 36 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
modules/observers/perf_observer.cc
View File

@ -7,13 +7,26 @@
#include <random>
#include <time.h>
#ifdef _WIN32
#include <windows.h>
#endif
#include "caffe2/core/common.h"
#include "caffe2/core/init.h"
#include "caffe2/core/operator.h"
#if defined(TARGET_OS_MAC) || \
defined(TARGET_OS_IPHONE) || \
defined(TARGET_IPHONE_SIMULATOR)
#define _APPLE 1
#endif
#ifdef _WIN32
#include <windows.h>
#endif
#ifdef _APPLE
#include <mach/mach_time.h>
#include <sys/time.h>
#include <sys/resource.h>
#endif
#ifndef C10_MOBILE
C10_DEFINE_int64(
aiBench_netInitSampleRate,
@ -78,7 +91,7 @@ double getTicksPerMillisecond() {
return static_cast<double>(ticks_per_sec.QuadPart) / 1000.0;
}
#else
#elif !defined _APPLE
double getClockTimeMilliseconds(clockid_t clk_id) {
int result;
struct timespec tp;
@ -102,6 +115,15 @@ double getWallClockTimeMilliseconds() {
}
return 0.0;
#elif defined _APPLE
static mach_timebase_info_data_t info;
if (info.denom == 0) {
mach_timebase_info(&info);
}
uint64_t now = mach_absolute_time();
now = now * info.numer / info.denom; // convert to nanoseconds
return now / 1000000.0;
#else
return getClockTimeMilliseconds(CLOCK_MONOTONIC);
#endif
@ -130,6 +152,16 @@ double getCpuTimeMilliseconds() {
}
return 0.0;
#elif defined _APPLE
struct rusage ru;
if (getrusage(RUSAGE_SELF, &ru)) {
return 0.0;
}
return ru.ru_utime.tv_sec * 1000.0
+ ru.ru_utime.tv_usec / 1000.0
+ ru.ru_stime.tv_sec * 1000.0
+ ru.ru_stime.tv_usec / 1000.0;
#else
return getClockTimeMilliseconds(CLOCK_PROCESS_CPUTIME_ID);
#endif