gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Code Quality: some cleanups

Browse Source
This commit is contained in:
Philippe Tillet
2017-05-11 16:11:40 -07:00
parent 3ff543151c
commit 9212ab5a3d
40 changed files with 112 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lib/driver/buffer.cpp
View File

@@ -34,11 +34,17 @@ namespace driver
{
Buffer::Buffer(Context const & /*context*/, size_t size) : size_(size)
{ dispatch::cuMemAlloc(&*cu_, size); }
Buffer::Buffer(Context const & context, size_t size) : context_(context), size_(size)
{
ContextSwitcher ctx_switch(context_);
dispatch::cuMemAlloc(&*cu_, size);
}
void Buffer::set_zero(Stream const & queue)
{ dispatch::cuMemsetD8Async(*cu_, 0, size_, queue); }
{
ContextSwitcher ctx_switch(context_);
dispatch::cuMemsetD8Async(*cu_, 0, size_, queue);
}
Handle<CUdeviceptr> const & Buffer::cu() const
{ return cu_; }