diff --git a/notebooks/ProjectMockup.ipynb b/notebooks/ProjectMockup.ipynb new file mode 100644 index 00000000..e1070353 --- /dev/null +++ b/notebooks/ProjectMockup.ipynb @@ -0,0 +1,249 @@ +{ + "cells": [ + { + "cell_type": "markdown", + "id": "1c4fd5b5-1b9e-4a8e-82b3-6ab214c203ea", + "metadata": {}, + "source": [ + "## Project Mockup" + ] + }, + { + "cell_type": "code", + "execution_count": 1, + "id": "54a52e27-3d9d-4574-990b-329a47d4bf13", + "metadata": {}, + "outputs": [], + "source": [ + "struct S { double val = 1.0; };" + ] + }, + { + "cell_type": "code", + "execution_count": 2, + "id": "b9e80eda-8924-4f3c-bbc0-7b74ba9b9a36", + "metadata": {}, + "outputs": [], + "source": [ + "%%python\n", + "\n", + "python_vec = cppyy.gbl.std.vector(cppyy.gbl.S)(5)" + ] + }, + { + "cell_type": "code", + "execution_count": 3, + "id": "697760dd-200e-4fa6-85b2-f1786215eda3", + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "1.0\n" + ] + } + ], + "source": [ + "%%python\n", + "\n", + "print(python_vec[3].val)" + ] + }, + { + "cell_type": "code", + "execution_count": 4, + "id": "d581f6cc-2737-4c11-b37a-25bb5296936a", + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "<__main__.Derived object at 0x7fffdf639e40>\n" + ] + }, + { + "name": "stderr", + "output_type": "stream", + "text": [ + ":1: RuntimeWarning: class \"S\" has no virtual destructor\n" + ] + } + ], + "source": [ + "%%python\n", + "\n", + "class Derived(cppyy.gbl.S):\n", + " def __init__(self):\n", + " val = 0\n", + "res = Derived()\n", + "print(res)" + ] + }, + { + "cell_type": "code", + "execution_count": 5, + "id": "f20feaa6-62a2-4805-853b-268b1c1832ac", + "metadata": {}, + "outputs": [], + "source": [ + "__global__ void arr_sum(int n, double *x, double *sum) {\n", + " for(int i = 0; i < n; i++)\n", + " *sum +=x[i];\n", + "}" + ] + }, + { + "cell_type": "code", + "execution_count": 6, + "id": "cfd92edb-41e0-4bb8-b7a1-852782964423", + "metadata": {}, + "outputs": [], + "source": [ + "int n = 5;\n", + "double h_sum;\n", + "double *x = new double[n];" + ] + }, + { + "cell_type": "code", + "execution_count": 7, + "id": "90480051-7a57-4145-85c4-ca3bdc8e101f", + "metadata": {}, + "outputs": [], + "source": [ + "void setData(const std::vector& a) {\n", + " int i = 0;\n", + " for(auto &s : a) {\n", + " x[i] = s.val;\n", + " i++;\n", + " }\n", + "}" + ] + }, + { + "cell_type": "code", + "execution_count": 8, + "id": "9d9913b7-db39-4c76-8939-efbca9256143", + "metadata": {}, + "outputs": [], + "source": [ + "%%python\n", + "\n", + "data_list = [1.0, 2.0, 3.0, 4.0, 5.0]\n", + "for c, i in enumerate(data_list):\n", + " python_vec[c].val = i\n" + ] + }, + { + "cell_type": "code", + "execution_count": 9, + "id": "846901c7-0cbb-4978-9efe-7f5870b939a7", + "metadata": {}, + "outputs": [], + "source": [ + "%%python\n", + "\n", + "cppyy.gbl.setData(python_vec)\n" + ] + }, + { + "cell_type": "code", + "execution_count": 10, + "id": "f57c46fa-addb-4d80-8cc3-29a464911fdc", + "metadata": {}, + "outputs": [], + "source": [ + "double *d_x, *d_sum;\n", + "cudaMalloc((void **)&d_x, n * sizeof(double));\n", + "cudaMalloc((void **)&d_sum, sizeof(double));" + ] + }, + { + "cell_type": "code", + "execution_count": 11, + "id": "d617ee2c-d9b7-4c4f-8ef2-051df37b2737", + "metadata": {}, + "outputs": [], + "source": [ + "cudaMemcpy(d_x, x, n * sizeof(double), cudaMemcpyHostToDevice);\n", + "cudaMemcpy(d_sum, &h_sum, sizeof(double), cudaMemcpyHostToDevice);" + ] + }, + { + "cell_type": "code", + "execution_count": 12, + "id": "8417b211-e7f9-448d-969a-27fc0eb32697", + "metadata": {}, + "outputs": [], + "source": [ + "arr_sum<<<1, 1>>>(n, d_x, d_sum);" + ] + }, + { + "cell_type": "code", + "execution_count": 13, + "id": "4374b687-31a5-4b85-8d66-e738956814b4", + "metadata": {}, + "outputs": [], + "source": [ + "cudaMemcpy(&h_sum, d_sum, sizeof(double), cudaMemcpyDeviceToHost);" + ] + }, + { + "cell_type": "code", + "execution_count": 14, + "id": "8198bfdf-0ff7-4e27-ba6b-087833055794", + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Sum: 15\n" + ] + } + ], + "source": [ + "std::cout << \"Sum: \" << h_sum << std::endl;" + ] + }, + { + "cell_type": "code", + "execution_count": 15, + "id": "33ffcf05-58cc-4212-aef7-4a5813fdc178", + "metadata": {}, + "outputs": [], + "source": [ + "delete[] x;\n", + "cudaFree(d_x);\n", + "cudaFree(d_sum);" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "add835ef-6196-47d8-be75-14b872438de1", + "metadata": {}, + "outputs": [], + "source": [] + } + ], + "metadata": { + "kernelspec": { + "display_name": "CUDA (C++17)", + "language": "CUDA", + "name": "cuda-xcpp17" + }, + "language_info": { + "codemirror_mode": "text/x-c++src", + "file_extension": ".cpp", + "mimetype": "text/x-c++src", + "name": "c++", + "version": "17" + } + }, + "nbformat": 4, + "nbformat_minor": 5 +}