mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
7b97eaeba5
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/30906 Add mobile module observer to measure performance of each method run. ghstack-source-id: 95120194 Test Plan: Run pytext model through BI cloaking flow on lite-interpreter and verify logs are sent: 1. buck install -r fb4a 2. Go to internal setting and find MobileConfig, search for android_bi_infra_cloaking_iab_models and set the following params: a. sample_rate: 1.0 b. enabled: true c. use_bytedoc_pytorch_model: true d. use_bytedoc_caffe2_model: false e. use_full_jit: false 3. Go back to new feed and scroll down until find an ads which will direct you to offsite webpage; 4. Click on the ads, wait for the offsite ads loads; 5. Click back to news feed; 6. Go to scuba table: https://fburl.com/scuba/4fghwp0b and see all the operator runs have been logged: {F223456981} Reviewed By: ljk53 Differential Revision: D18702116 fbshipit-source-id: a9f07eee684e3022cef5ba3c5934f30f20192a85
11 lines
188 B
C++
11 lines
188 B
C++
#include <torch/csrc/jit/mobile/observer.h>
|
|
|
|
namespace torch {
|
|
|
|
MobileObserverConfig& observerConfig() {
|
|
static MobileObserverConfig instance;
|
|
return instance;
|
|
}
|
|
|
|
} // namespace torch
|