Transfer files from RAD repository

tests/sos_tests/asym_alloc.cpp

@@ -0,0 +1,88 @@
+/*
+ *  Copyright (c) 2017 Intel Corporation. All rights reserved.
+ *  This software is available to you under the BSD license below:
+ *
+ *      Redistribution and use in source and binary forms, with or
+ *      without modification, are permitted provided that the following
+ *      conditions are met:
+ *
+ *      - Redistributions of source code must retain the above
+ *        copyright notice, this list of conditions and the following
+ *        disclaimer.
+ *
+ *      - Redistributions in binary form must reproduce the above
+ *        copyright notice, this list of conditions and the following
+ *        disclaimer in the documentation and/or other materials
+ *        provided with the distribution.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+ * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+ * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+ * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+/* Synopsis: Test that a single asymmetric allocation works correctly.
+ *
+ * This semantic is provided in OpenSHMEM 1.1 and some versions of Cray SHMEM.
+ * It was removed from OpenSHMEM in 1.2, but we maintain it for backward
+ * compatibility.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include <roc_shmem/roc_shmem.hpp>
+
+using namespace rocshmem;
+
+long bufsize;
+
+int main(int argc, char **argv) {
+  int *buf, *buf_in;
+  int me, npes, i, target;
+
+  roc_shmem_init();
+  me = roc_shmem_my_pe();
+  npes = roc_shmem_n_pes();
+
+  /* Each PE allocates space for "me + 1" integers */
+  bufsize = me + 1;
+  buf = (int *)roc_shmem_malloc(sizeof(int) * bufsize);
+
+  if (NULL == buf) roc_shmem_global_exit(1);
+
+  for (i = 0; i < bufsize; i++) buf[i] = -1;
+
+  roc_shmem_barrier_all();
+
+  /* Write to neighbor's buffer */
+  target = (me + 1) % npes;
+  buf_in = (int *)malloc(sizeof(int) * (target + 1));
+  if (!buf_in) {
+    fprintf(stderr, "ERR - null buf_in pointer\n");
+    roc_shmem_global_exit(1);
+  }
+
+  for (i = 0; i < target + 1; i++) buf_in[i] = target;
+
+  roc_shmem_int_put(buf, buf_in, target + 1, target);
+
+  roc_shmem_barrier_all();
+
+  /* Validate data was written correctly */
+  for (i = 0; i < me + 1; i++) {
+    if (buf[i] != me) {
+      printf("Error [%3d]: buf[%d] == %d, expected %d\n", me, i, buf[i], me);
+      roc_shmem_global_exit(2);
+    }
+  }
+
+  free(buf_in);
+  roc_shmem_free(buf);
+  roc_shmem_finalize();
+  return 0;
+}