blob: e05fd06f7faa32bbf84e525f072d5a17b92585e5 [file] [log] [blame]
<!DOCTYPE html>
<html class="writer-html5" lang="en" >
<head>
<meta charset="utf-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>CUDA Integration &mdash; Apache Arrow v2.0.0</title>
<link rel="stylesheet" href="../../_static/css/theme.css" type="text/css" />
<link rel="stylesheet" href="../../_static/pygments.css" type="text/css" />
<!--[if lt IE 9]>
<script src="../../_static/js/html5shiv.min.js"></script>
<![endif]-->
<script type="text/javascript" id="documentation_options" data-url_root="../../" src="../../_static/documentation_options.js"></script>
<script src="../../_static/jquery.js"></script>
<script src="../../_static/underscore.js"></script>
<script src="../../_static/doctools.js"></script>
<script src="../../_static/language_data.js"></script>
<script type="text/javascript" src="../../_static/js/theme.js"></script>
<link rel="canonical" href="https://arrow.apache.org/docs/python/api/cuda.html" />
<link rel="index" title="Index" href="../../genindex.html" />
<link rel="search" title="Search" href="../../search.html" />
<link rel="next" title="pyarrow.cuda.Context" href="../generated/pyarrow.cuda.Context.html" />
<link rel="prev" title="pyarrow.plasma.PlasmaBuffer" href="../generated/pyarrow.plasma.PlasmaBuffer.html" />
<!-- Matomo -->
<script>
var _paq = window._paq = window._paq || [];
/* tracker methods like "setCustomDimension" should be called before "trackPageView" */
_paq.push(["setDoNotTrack", true]);
_paq.push(["disableCookies"]);
_paq.push(['trackPageView']);
_paq.push(['enableLinkTracking']);
(function() {
var u="https://analytics.apache.org/";
_paq.push(['setTrackerUrl', u+'matomo.php']);
_paq.push(['setSiteId', '20']);
var d=document, g=d.createElement('script'), s=d.getElementsByTagName('script')[0];
g.async=true; g.src=u+'matomo.js'; s.parentNode.insertBefore(g,s);
})();
</script>
<!-- End Matomo Code -->
</head>
<body class="wy-body-for-nav">
<div class="wy-grid-for-nav">
<nav data-toggle="wy-nav-shift" class="wy-nav-side">
<div class="wy-side-scroll">
<div class="wy-side-nav-search" >
<a href="../../index.html" class="icon icon-home" alt="Documentation Home"> Apache Arrow
</a>
<div class="version">
2.0.0
</div>
<div role="search">
<form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">
<input type="text" name="q" placeholder="Search docs" />
<input type="hidden" name="check_keywords" value="yes" />
<input type="hidden" name="area" value="default" />
</form>
</div>
</div>
<div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
<p class="caption"><span class="caption-text">Specifications and Protocols</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../format/Versioning.html">Format Versioning and Stability</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../format/Columnar.html">Arrow Columnar Format</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../format/Flight.html">Arrow Flight RPC</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../format/Integration.html">Integration Testing</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../format/CDataInterface.html">The Arrow C data interface</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../format/CStreamInterface.html">The Arrow C stream interface</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../format/Other.html">Other Data Structures</a></li>
</ul>
<p class="caption"><span class="caption-text">Libraries</span></p>
<ul class="current">
<li class="toctree-l1"><a class="reference internal" href="../../status.html">Implementation Status</a></li>
<li class="toctree-l1"><a class="reference external" href="https://arrow.apache.org/docs/c_glib/">C/GLib</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../cpp/index.html">C++</a></li>
<li class="toctree-l1"><a class="reference external" href="https://github.com/apache/arrow/blob/master/csharp/README.md">C#</a></li>
<li class="toctree-l1"><a class="reference external" href="https://godoc.org/github.com/apache/arrow/go/arrow">Go</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../java/index.html">Java</a></li>
<li class="toctree-l1"><a class="reference external" href="https://arrow.apache.org/docs/js/">JavaScript</a></li>
<li class="toctree-l1"><a class="reference external" href="https://github.com/apache/arrow/blob/master/matlab/README.md">MATLAB</a></li>
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">Python</a><ul class="current">
<li class="toctree-l2"><a class="reference internal" href="../install.html">Installing PyArrow</a></li>
<li class="toctree-l2"><a class="reference internal" href="../memory.html">Memory and IO Interfaces</a></li>
<li class="toctree-l2"><a class="reference internal" href="../data.html">Data Types and In-Memory Data Model</a></li>
<li class="toctree-l2"><a class="reference internal" href="../compute.html">Compute Functions</a></li>
<li class="toctree-l2"><a class="reference internal" href="../ipc.html">Streaming, Serialization, and IPC</a></li>
<li class="toctree-l2"><a class="reference internal" href="../filesystems.html">Filesystem Interface</a></li>
<li class="toctree-l2"><a class="reference internal" href="../filesystems_deprecated.html">Filesystem Interface (legacy)</a></li>
<li class="toctree-l2"><a class="reference internal" href="../plasma.html">The Plasma In-Memory Object Store</a></li>
<li class="toctree-l2"><a class="reference internal" href="../numpy.html">NumPy Integration</a></li>
<li class="toctree-l2"><a class="reference internal" href="../pandas.html">Pandas Integration</a></li>
<li class="toctree-l2"><a class="reference internal" href="../timestamps.html">Timestamps</a></li>
<li class="toctree-l2"><a class="reference internal" href="../csv.html">Reading CSV files</a></li>
<li class="toctree-l2"><a class="reference internal" href="../feather.html">Feather File Format</a></li>
<li class="toctree-l2"><a class="reference internal" href="../json.html">Reading JSON files</a></li>
<li class="toctree-l2"><a class="reference internal" href="../parquet.html">Reading and Writing the Apache Parquet Format</a></li>
<li class="toctree-l2"><a class="reference internal" href="../dataset.html">Tabular Datasets</a></li>
<li class="toctree-l2"><a class="reference internal" href="../cuda.html">CUDA Integration</a></li>
<li class="toctree-l2"><a class="reference internal" href="../extending_types.html">Extending pyarrow</a></li>
<li class="toctree-l2"><a class="reference internal" href="../extending.html">Using pyarrow from C++ and Cython Code</a></li>
<li class="toctree-l2 current"><a class="reference internal" href="../api.html">API Reference</a><ul class="current">
<li class="toctree-l3"><a class="reference internal" href="datatypes.html">Data Types and Schemas</a></li>
<li class="toctree-l3"><a class="reference internal" href="arrays.html">Arrays and Scalars</a></li>
<li class="toctree-l3"><a class="reference internal" href="memory.html">Buffers and Memory</a></li>
<li class="toctree-l3"><a class="reference internal" href="compute.html">Compute Functions</a></li>
<li class="toctree-l3"><a class="reference internal" href="files.html">Streams and File Access</a></li>
<li class="toctree-l3"><a class="reference internal" href="tables.html">Tables and Tensors</a></li>
<li class="toctree-l3"><a class="reference internal" href="ipc.html">Serialization and IPC</a></li>
<li class="toctree-l3"><a class="reference internal" href="flight.html">Arrow Flight</a></li>
<li class="toctree-l3"><a class="reference internal" href="formats.html">Tabular File Formats</a></li>
<li class="toctree-l3"><a class="reference internal" href="filesystems.html">Filesystems</a></li>
<li class="toctree-l3"><a class="reference internal" href="dataset.html">Dataset</a></li>
<li class="toctree-l3"><a class="reference internal" href="plasma.html">Plasma In-Memory Object Store</a></li>
<li class="toctree-l3 current"><a class="current reference internal" href="#">CUDA Integration</a><ul>
<li class="toctree-l4"><a class="reference internal" href="#cuda-contexts">CUDA Contexts</a></li>
<li class="toctree-l4"><a class="reference internal" href="#cuda-buffers">CUDA Buffers</a></li>
<li class="toctree-l4"><a class="reference internal" href="#serialization-and-ipc">Serialization and IPC</a></li>
</ul>
</li>
<li class="toctree-l3"><a class="reference internal" href="misc.html">Miscellaneous</a></li>
</ul>
</li>
<li class="toctree-l2"><a class="reference internal" href="../getting_involved.html">Getting Involved</a></li>
<li class="toctree-l2"><a class="reference internal" href="../benchmarks.html">Benchmarks</a></li>
</ul>
</li>
<li class="toctree-l1"><a class="reference external" href="https://arrow.apache.org/docs/r/">R</a></li>
<li class="toctree-l1"><a class="reference external" href="https://github.com/apache/arrow/blob/master/ruby/README.md">Ruby</a></li>
<li class="toctree-l1"><a class="reference external" href="https://docs.rs/crate/arrow/">Rust</a></li>
</ul>
<p class="caption"><span class="caption-text">Development</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../developers/contributing.html">Contributing to Apache Arrow</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../developers/cpp/index.html">C++ Development</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../developers/python.html">Python Development</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../developers/archery.html">Daily Development using Archery</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../developers/crossbow.html">Packaging and Testing with Crossbow</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../developers/docker.html">Running Docker Builds</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../developers/benchmarks.html">Benchmarks</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../developers/documentation.html">Building the Documentation</a></li>
</ul>
</div>
</div>
</nav>
<section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
<nav class="wy-nav-top" aria-label="top navigation">
<i data-toggle="wy-nav-top" class="fa fa-bars"></i>
<a href="../../index.html">Apache Arrow</a>
</nav>
<div class="wy-nav-content">
<div class="rst-content">
<div role="navigation" aria-label="breadcrumbs navigation">
<ul class="wy-breadcrumbs">
<li><a href="../../index.html" class="icon icon-home"></a> &raquo;</li>
<li><a href="../index.html">Python bindings</a> &raquo;</li>
<li><a href="../api.html">API Reference</a> &raquo;</li>
<li>CUDA Integration</li>
<li class="wy-breadcrumbs-aside">
<a href="../../_sources/python/api/cuda.rst.txt" rel="nofollow"> View page source</a>
</li>
</ul>
<hr/>
</div>
<div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
<div itemprop="articleBody">
<div class="section" id="cuda-integration">
<h1>CUDA Integration<a class="headerlink" href="#cuda-integration" title="Permalink to this headline"></a></h1>
<div class="section" id="cuda-contexts">
<h2>CUDA Contexts<a class="headerlink" href="#cuda-contexts" title="Permalink to this headline"></a></h2>
<table class="longtable docutils align-default">
<colgroup>
<col style="width: 10%" />
<col style="width: 90%" />
</colgroup>
<tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.Context.html#pyarrow.cuda.Context" title="pyarrow.cuda.Context"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Context</span></code></a></p></td>
<td><p>CUDA driver context.</p></td>
</tr>
</tbody>
</table>
</div>
<div class="section" id="cuda-buffers">
<h2>CUDA Buffers<a class="headerlink" href="#cuda-buffers" title="Permalink to this headline"></a></h2>
<table class="longtable docutils align-default">
<colgroup>
<col style="width: 10%" />
<col style="width: 90%" />
</colgroup>
<tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.CudaBuffer.html#pyarrow.cuda.CudaBuffer" title="pyarrow.cuda.CudaBuffer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">CudaBuffer</span></code></a></p></td>
<td><p>An Arrow buffer with data located in a GPU device.</p></td>
</tr>
<tr class="row-even"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.new_host_buffer.html#pyarrow.cuda.new_host_buffer" title="pyarrow.cuda.new_host_buffer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">new_host_buffer</span></code></a></p></td>
<td><p>Return buffer with CUDA-accessible memory on CPU host</p></td>
</tr>
<tr class="row-odd"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.HostBuffer.html#pyarrow.cuda.HostBuffer" title="pyarrow.cuda.HostBuffer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">HostBuffer</span></code></a></p></td>
<td><p>Device-accessible CPU memory created using cudaHostAlloc.</p></td>
</tr>
<tr class="row-even"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.BufferReader.html#pyarrow.cuda.BufferReader" title="pyarrow.cuda.BufferReader"><code class="xref py py-obj docutils literal notranslate"><span class="pre">BufferReader</span></code></a></p></td>
<td><p>File interface for zero-copy read from CUDA buffers.</p></td>
</tr>
<tr class="row-odd"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.BufferWriter.html#pyarrow.cuda.BufferWriter" title="pyarrow.cuda.BufferWriter"><code class="xref py py-obj docutils literal notranslate"><span class="pre">BufferWriter</span></code></a></p></td>
<td><p>File interface for writing to CUDA buffers.</p></td>
</tr>
</tbody>
</table>
</div>
<div class="section" id="serialization-and-ipc">
<h2>Serialization and IPC<a class="headerlink" href="#serialization-and-ipc" title="Permalink to this headline"></a></h2>
<table class="longtable docutils align-default">
<colgroup>
<col style="width: 10%" />
<col style="width: 90%" />
</colgroup>
<tbody>
<tr class="row-odd"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.serialize_record_batch.html#pyarrow.cuda.serialize_record_batch" title="pyarrow.cuda.serialize_record_batch"><code class="xref py py-obj docutils literal notranslate"><span class="pre">serialize_record_batch</span></code></a></p></td>
<td><p>Write record batch message to GPU device memory</p></td>
</tr>
<tr class="row-even"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.read_record_batch.html#pyarrow.cuda.read_record_batch" title="pyarrow.cuda.read_record_batch"><code class="xref py py-obj docutils literal notranslate"><span class="pre">read_record_batch</span></code></a></p></td>
<td><p>Construct RecordBatch referencing IPC message located on CUDA device.</p></td>
</tr>
<tr class="row-odd"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.read_message.html#pyarrow.cuda.read_message" title="pyarrow.cuda.read_message"><code class="xref py py-obj docutils literal notranslate"><span class="pre">read_message</span></code></a></p></td>
<td><p>Read Arrow IPC message located on GPU device</p></td>
</tr>
<tr class="row-even"><td><p><a class="reference internal" href="../generated/pyarrow.cuda.IpcMemHandle.html#pyarrow.cuda.IpcMemHandle" title="pyarrow.cuda.IpcMemHandle"><code class="xref py py-obj docutils literal notranslate"><span class="pre">IpcMemHandle</span></code></a></p></td>
<td><p>A serializable container for a CUDA IPC handle.</p></td>
</tr>
</tbody>
</table>
</div>
</div>
</div>
</div>
<footer>
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
<a href="../generated/pyarrow.cuda.Context.html" class="btn btn-neutral float-right" title="pyarrow.cuda.Context" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right"></span></a>
<a href="../generated/pyarrow.plasma.PlasmaBuffer.html" class="btn btn-neutral float-left" title="pyarrow.plasma.PlasmaBuffer" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left"></span> Previous</a>
</div>
<hr/>
<div role="contentinfo">
<p>
&copy; Copyright 2016-2019 Apache Software Foundation
</p>
</div>
Built with <a href="http://sphinx-doc.org/">Sphinx</a> using a
<a href="https://github.com/rtfd/sphinx_rtd_theme">theme</a>
provided by <a href="https://readthedocs.org">Read the Docs</a>.
</footer>
</div>
</div>
</section>
</div>
<script type="text/javascript">
jQuery(function () {
SphinxRtdTheme.Navigation.enable(true);
});
</script>
<script type="text/javascript" src="/docs/_static/versionwarning.js"></script></body>
</html>