MCPcopy Create free account
hub / github.com/dmlc/xgboost / get_current_use_rmm_flag

Function get_current_use_rmm_flag

tests/python/test_config.py:24–25  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

22@pytest.mark.parametrize("use_rmm", [False, True])
23def test_global_config_use_rmm(use_rmm):
24 def get_current_use_rmm_flag():
25 return xgb.get_config()["use_rmm"]
26
27 old_use_rmm_flag = get_current_use_rmm_flag()
28 with xgb.config_context(use_rmm=use_rmm):

Callers 1

Calls

no outgoing calls

Tested by

no test coverage detected