MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / IsUseCUDA

Method IsUseCUDA

paddle/fluid/framework/compiled_program.cc:334–336  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

332};
333
334bool CompiledProgramPrivate::IsUseCUDA(DeviceType use_device) {
335 return use_device == p::kCUDA;
336}
337
338ir::Graph *CompiledProgramPrivate::ApplyMemoryOptimizePass(ir::Graph *graph) {
339 std::vector<ir::LastLiveOpsOfVars> last_live_ops_of_vars;

Callers 4

BCastParamsToDevicesMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected