MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / set_prealloc_config

Method set_prealloc_config

src/core/impl/comp_node/comp_node.cpp:412–423  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

410}
411
412void CompNode::set_prealloc_config(
413 size_t alignment, size_t min_req, size_t max_overhead, double growth_factor,
414 DeviceType device_type) {
415 switch (device_type) {
416 case DeviceType::CUDA:
417 CudaCompNode::set_prealloc_config(
418 alignment, min_req, max_overhead, growth_factor);
419 break;
420 default:
421 mgb_log_warn("unsupported device type for set_prealloc_config");
422 };
423}
424
425CompNode::DeviceProperties CompNode::get_device_prop(int dev, DeviceType device_type) {
426 switch (device_type) {

Callers

nothing calls this directly

Calls 1

set_prealloc_configFunction · 0.85

Tested by

no test coverage detected