From 955d8375ef5de5460811037c14ffaffabc8d47a5 Mon Sep 17 00:00:00 2001 From: Tim Date: Tue, 28 Jan 2025 04:50:34 +0000 Subject: [PATCH 1/3] support reservation arn --- runtime/common/BraketExecutor.h | 1 + .../default/rest/helpers/braket/BraketExecutor.cpp | 14 +++++++++++++- 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/runtime/common/BraketExecutor.h b/runtime/common/BraketExecutor.h index 258e5c9425..abd451307d 100644 --- a/runtime/common/BraketExecutor.h +++ b/runtime/common/BraketExecutor.h @@ -55,6 +55,7 @@ class BraketExecutor : public Executor { std::shared_future defaultBucketFuture; char const *jobToken; + char const *reservationArn; std::chrono::microseconds pollingInterval = std::chrono::milliseconds{100}; diff --git a/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp b/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp index 7a46a7b0da..a181c9b713 100644 --- a/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp +++ b/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp @@ -12,6 +12,8 @@ #include #include #include +#include +#include #include #include @@ -100,7 +102,9 @@ void tryCreateBucket(Aws::S3Crt::S3CrtClient &client, std::string const ®ion, namespace cudaq { BraketExecutor::BraketExecutor() - : api(options), jobToken(std::getenv("AMZN_BRAKET_JOB_TOKEN")) {} + : api(options), + jobToken(std::getenv("AMZN_BRAKET_JOB_TOKEN")), + reservationArn(std::getenv("AMZN_BRAKET_RESERVATION_TIME_WINDOW_ARN")) {} /// @brief Set the server helper void BraketExecutor::setServerHelper(ServerHelper *helper) { @@ -190,6 +194,14 @@ BraketExecutor::execute(std::vector &codesToExecute, req.SetShots(message["shots"]); if (jobToken) req.SetJobToken(jobToken); + + if (reservationArn) { + Aws::Braket::Model::Association assoc; + assoc.SetArn(reservationArn); + assoc.SetType(Aws::Braket::Model::AssociationType::RESERVATION_TIME_WINDOW_ARN); + req.AddAssociations(std::move(assoc)); + } + req.SetOutputS3Bucket(defaultBucket); req.SetOutputS3KeyPrefix(defaultPrefix); From 1c5e5bd9b0d501500b8eb5e10343d469898d63b5 Mon Sep 17 00:00:00 2001 From: Tim Date: Tue, 28 Jan 2025 14:43:57 +0000 Subject: [PATCH 2/3] DCO Remediation Commit for Tim I, Tim , hereby add my Signed-off-by to this commit: 955d8375ef5de5460811037c14ffaffabc8d47a5 Signed-off-by: Tim From db6311c6ee9f867f58294902712c5b174ae7f35b Mon Sep 17 00:00:00 2001 From: Tim Date: Fri, 7 Feb 2025 23:13:05 +0000 Subject: [PATCH 3/3] fix: run clang formatter Signed-off-by: Tim --- .../default/rest/helpers/braket/BraketExecutor.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp b/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp index a181c9b713..2ea8eea8ba 100644 --- a/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp +++ b/runtime/cudaq/platform/default/rest/helpers/braket/BraketExecutor.cpp @@ -9,11 +9,11 @@ #include "common/BraketExecutor.h" #include "common/BraketServerHelper.h" +#include +#include #include #include #include -#include -#include #include #include @@ -102,8 +102,7 @@ void tryCreateBucket(Aws::S3Crt::S3CrtClient &client, std::string const ®ion, namespace cudaq { BraketExecutor::BraketExecutor() - : api(options), - jobToken(std::getenv("AMZN_BRAKET_JOB_TOKEN")), + : api(options), jobToken(std::getenv("AMZN_BRAKET_JOB_TOKEN")), reservationArn(std::getenv("AMZN_BRAKET_RESERVATION_TIME_WINDOW_ARN")) {} /// @brief Set the server helper @@ -198,7 +197,8 @@ BraketExecutor::execute(std::vector &codesToExecute, if (reservationArn) { Aws::Braket::Model::Association assoc; assoc.SetArn(reservationArn); - assoc.SetType(Aws::Braket::Model::AssociationType::RESERVATION_TIME_WINDOW_ARN); + assoc.SetType( + Aws::Braket::Model::AssociationType::RESERVATION_TIME_WINDOW_ARN); req.AddAssociations(std::move(assoc)); }