upload our code

SYSU-SCC · May 5, 2021 · ead1524 · ead1524
1 parent d2c19e8
commit ead1524
Show file tree

Hide file tree

Showing 25 changed files with 4,573 additions and 104 deletions.
diff --git a/QuEST/CMakeLists.txt b/QuEST/CMakeLists.txt
@@ -49,11 +49,11 @@ message(STATUS "MPI distribution is ${DISTRIBUTED}")
 
 # ----- FATAL -----------------------------------------------------------------
 
-if (${DISTRIBUTED} AND ${GPUACCELERATED})
-    message(FATAL_ERROR "DISTRIBUTED=${DISTRIBUTED} and \
-        GPUACCELERATED=${GPUACCELERATED} set but \
-        distributed GPU acceleration not supported. Aborting")
-endif()
+# if (${DISTRIBUTED} AND ${GPUACCELERATED})
+#    message(FATAL_ERROR "DISTRIBUTED=${DISTRIBUTED} and \
+#        GPUACCELERATED=${GPUACCELERATED} set but \
+#        distributed GPU acceleration not supported. Aborting")
+# endif()
 
 if ( NOT(${PRECISION} EQUAL 1) AND
      NOT(${PRECISION} EQUAL 2) AND 
@@ -71,6 +71,13 @@ endif()
 
 # ----- WARNINGS --------------------------------------------------------------
 
+if (${DISTRIBUTED} AND ${GPUACCELERATED})
+    message(WARNING "DISTRIBUTED=${DISTRIBUTED} and \
+        GPUACCELERATED=${GPUACCELERATED} set. \
+        Distributed GPU acceleration is supported by \
+        Student Cluster Competition Team @ Sun Yat-sen University.")
+endif()
+
 if (${GPUACCELERATED} AND ${MULTITHREADED})
     message(WARNING "MULTITHREADED=${MULTITHREADED} and \
         GPUACCELERATED=${GPUACCELERATED} set but GPU \

diff --git a/QuEST/include/QuEST.h b/QuEST/include/QuEST.h
@@ -108,7 +108,7 @@ typedef struct Qureg
 
     //! Storage for generated QASM output
     QASMLogger* qasmLog;
-
+    void *rhandle;
 } Qureg;
 
 /** Information about the environment the program is running in.
@@ -118,6 +118,7 @@ typedef struct QuESTEnv
 {
     int rank;
     int numRanks;
+    void* ehandle;
 } QuESTEnv;
 
 

diff --git a/QuEST/src/GPU/CMakeLists.txt b/QuEST/src/GPU/CMakeLists.txt
@@ -1,6 +1,11 @@
-set(QuEST_SRC_ARCHITECTURE_DEPENDENT
-    ${CMAKE_CURRENT_SOURCE_DIR}/QuEST_gpu.cu
-    PARENT_SCOPE
-)
-
-
+if (DISTRIBUTED)
+    set(QuEST_SRC_ARCHITECTURE_DEPENDENT
+        ${CMAKE_CURRENT_SOURCE_DIR}/QuEST_gpu_distributed.cu
+        PARENT_SCOPE
+    )
+else()
+    set(QuEST_SRC_ARCHITECTURE_DEPENDENT
+        ${CMAKE_CURRENT_SOURCE_DIR}/QuEST_gpu.cu
+        PARENT_SCOPE
+    )
+endif()