mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
aeb55225e0
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59355 Add a `CheckKnob()` function for doing run-time checks of feature roll-out knobs. This provides an API for safely controlling the roll-out of new functionality in the code. Test Plan: Included some basic unit tests. Reviewed By: voznesenskym Differential Revision: D26536430 fbshipit-source-id: 2e53234c6d9ce624848fc8b2c76f6833f344f48b
27 lines
532 B
C++
27 lines
532 B
C++
#pragma once
|
|
|
|
// This file contains functions for checking rollout knobs to enable staged
|
|
// roll out of specific code functionality.
|
|
|
|
#include <memory>
|
|
|
|
#include <c10/util/string_view.h>
|
|
|
|
namespace caffe2 {
|
|
|
|
/**
|
|
* Check an arbitrary knob by name.
|
|
*/
|
|
bool CheckKnob(c10::string_view name);
|
|
|
|
/*
|
|
* The following are functions for checking specific known knob values.
|
|
*
|
|
* These APIs are more efficient than checking by name.
|
|
*/
|
|
|
|
// An example knob, just for use in unit tests.
|
|
bool CheckKnobExampleKnob();
|
|
|
|
} // namespace caffe2
|