1
2 //
3 // This source file is part of appleseed.
4 // Visit https://appleseedhq.net/ for additional information and resources.
5 //
6 // This software is released under the MIT license.
7 //
8 // Copyright (c) 2019 Esteban Tovagliari, The appleseedhq Organization
9 //
10 // Permission is hereby granted, free of charge, to any person obtaining a copy
11 // of this software and associated documentation files (the "Software"), to deal
12 // in the Software without restriction, including without limitation the rights
13 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
14 // copies of the Software, and to permit persons to whom the Software is
15 // furnished to do so, subject to the following conditions:
16 //
17 // The above copyright notice and this permission notice shall be included in
18 // all copies or substantial portions of the Software.
19 //
20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
21 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
22 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
23 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
24 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
25 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
26 // THE SOFTWARE.
27 //
28
29 // Interface header.
30 #include "module.h"
31
32 // appleseed.foundation headers.
33 #include "foundation/cuda/exception.h"
34
35 // Standard headers.
36 #include <algorithm>
37 #include <cassert>
38
39 using namespace std;
40
41 namespace foundation
42 {
43
CUDAModule()44 CUDAModule::CUDAModule()
45 : m_module(nullptr)
46 {
47 }
48
CUDAModule(const char * filename)49 CUDAModule::CUDAModule(const char* filename)
50 {
51 check_cuda_result(cuModuleLoad(&m_module, filename));
52 }
53
CUDAModule(CUDAModule && rhs)54 CUDAModule::CUDAModule(CUDAModule&& rhs)
55 : m_module(rhs.m_module)
56 {
57 rhs.m_module = nullptr;
58 }
59
~CUDAModule()60 CUDAModule::~CUDAModule()
61 {
62 if (m_module)
63 cuModuleUnload(m_module);
64 }
65
operator =(CUDAModule && rhs)66 CUDAModule& CUDAModule::operator=(CUDAModule&& rhs)
67 {
68 CUDAModule tmp(move(rhs));
69 swap(m_module, tmp.m_module);
70 return *this;
71 }
72
get_function(const char * name)73 CUfunction CUDAModule::get_function(const char* name)
74 {
75 assert(m_module);
76
77 CUfunction f;
78 check_cuda_result(cuModuleGetFunction(&f, m_module, name));
79 return f;
80 }
81
82 } // namespace foundation
83