fix Thrust seg fault

Qiskit · Dec 8, 2023 · 259aecb · 259aecb
1 parent e5626df
commit 259aecb
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/src/simulators/density_matrix/densitymatrix.hpp b/src/simulators/density_matrix/densitymatrix.hpp
@@ -242,12 +242,12 @@ void DensityMatrix<data_t>::initialize_from_vector(list_t &&vec) {
 
 template <typename data_t>
 void DensityMatrix<data_t>::transpose() {
-  const size_t rows = BaseMatrix::num_rows();
+  const int_t rows = BaseMatrix::num_rows();
 #pragma omp parallel for if (BaseVector::num_qubits_ >                         \
                                  BaseVector::omp_threshold_ &&                 \
                              BaseVector::omp_threads_ > 1)                     \
     num_threads(BaseVector::omp_threads_)
-  for (uint_t i = 0; i < rows; i++) {
+  for (int_t i = 0; i < rows; i++) {
     for (uint_t j = i + 1; j < rows; j++) {
       const uint_t pos_a = i * rows + j;
       const uint_t pos_b = j * rows + i;

diff --git a/src/simulators/extended_stabilizer/extended_stabilizer_state.hpp b/src/simulators/extended_stabilizer/extended_stabilizer_state.hpp
@@ -69,7 +69,7 @@ class State : public QuantumState::State<chstate_t> {
   // Apply a sequence of operations to the cicuit. For each operation,
   // we loop over the terms in the decomposition in parallel
   template <typename InputIterator>
-  void apply_ops_stab(InputIterator first, InputIterator last,
+  void apply_ops(InputIterator first, InputIterator last,
                       ExperimentResult &result, RngEngine &rng,
                       bool final_ops = false);
 
@@ -341,11 +341,11 @@ bool State::check_measurement_opt(InputIterator first,
 //-------------------------------------------------------------------------
 void State::apply_op(const Operations::Op &op, ExperimentResult &result,
                      RngEngine &rng, bool final_op) {
-  apply_ops_stab(&op, &op + 1, result, rng, final_op);
+  apply_ops(&op, &op + 1, result, rng, final_op);
 }
 
 template <typename InputIterator>
-void State::apply_ops_stab(InputIterator first, InputIterator last,
+void State::apply_ops(InputIterator first, InputIterator last,
                            ExperimentResult &result, RngEngine &rng,
                            bool final_ops) {
   std::pair<bool, size_t> stabilizer_opts = check_stabilizer_opt(first, last);

diff --git a/src/simulators/statevector/qubitvector_thrust.hpp b/src/simulators/statevector/qubitvector_thrust.hpp
@@ -872,6 +872,7 @@ uint_t QubitVectorThrust<data_t>::chunk_setup(int chunk_bits, int num_qubits,
 
     if (chunk_manager_->chunk_bits() == chunk_bits &&
         chunk_manager_->num_qubits() == num_qubits) {
+      chunk_manager_->MapChunk(chunk_, 0);
       chunk_.set_chunk_index(chunk_index_);
       return num_local_chunks;
     }
@@ -903,6 +904,7 @@ uint_t QubitVectorThrust<data_t>::chunk_setup(int chunk_bits, int num_qubits,
 
   // mapping/setting chunk
   chunk_.set_chunk_index(chunk_index_);
+  chunk_manager_->MapChunk(chunk_, 0);
 
   return num_chunks_allocated;
 }
@@ -930,6 +932,7 @@ QubitVectorThrust<data_t>::chunk_setup(const QubitVectorThrust<data_t> &base,
 
   // mapping/setting chunk
   chunk_manager_ = base.chunk_manager_;
+  chunk_manager_->MapChunk(chunk_, 0);
 
   return 0;
 }