]>
git.proxmox.com Git - ceph.git/blob - ceph/src/arrow/ruby/red-arrow-cuda/test/test-cuda.rb
1 # Licensed to the Apache Software Foundation (ASF) under one
2 # or more contributor license agreements. See the NOTICE file
3 # distributed with this work for additional information
4 # regarding copyright ownership. The ASF licenses this file
5 # to you under the Apache License, Version 2.0 (the
6 # "License"); you may not use this file except in compliance
7 # with the License. You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing,
12 # software distributed under the License is distributed on an
13 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14 # KIND, either express or implied. See the License for the
15 # specific language governing permissions and limitations
18 class TestCUDA
< Test
::Unit::TestCase
20 @manager = ArrowCUDA
::DeviceManager.new
21 omit("At least one GPU is required") if @manager.n_devices
.zero
?
22 @context = @manager[0]
25 sub_test_case("BufferOutputStream") do
28 @buffer = ArrowCUDA
::Buffer.new(@context, 128)
32 ArrowCUDA
::BufferOutputStream.open(@buffer) do |stream
|
33 stream
.write("Hello World")
35 assert_equal("Hello World", @buffer.copy_to_host(0, 11).to_s
)