Internal change
PiperOrigin-RevId: 522014435
This commit is contained in:
parent
425a3ee3f6
commit
c5bd34ddb0
|
@ -119,7 +119,7 @@ GpuResources::~GpuResources() {
|
||||||
extern const GraphService<GpuResources> kGpuService;
|
extern const GraphService<GpuResources> kGpuService;
|
||||||
|
|
||||||
absl::Status GpuResources::PrepareGpuNode(CalculatorNode* node) {
|
absl::Status GpuResources::PrepareGpuNode(CalculatorNode* node) {
|
||||||
CHECK(ContainsKey(node->Contract().ServiceRequests(), kGpuService.key));
|
CHECK(node->Contract().ServiceRequests().contains(kGpuService.key));
|
||||||
std::string node_id = node->GetCalculatorState().NodeName();
|
std::string node_id = node->GetCalculatorState().NodeName();
|
||||||
std::string node_type = node->GetCalculatorState().CalculatorType();
|
std::string node_type = node->GetCalculatorState().CalculatorType();
|
||||||
std::string context_key;
|
std::string context_key;
|
||||||
|
|
Loading…
Reference in New Issue
Block a user