Commit 061bc56e authored by wsttiger's avatar wsttiger
Browse files

Added files needed for rocblas integration

parent c2b24de5
#ifndef MIGRAPH_GUARD_MIGRAPHLIB_ROCBLAS_HPP
#define MIGRAPH_GUARD_MIGRAPHLIB_ROCBLAS_HPP
#include <migraph/manage_ptr.hpp>
#include <migraph/operators.hpp>
#include <rocblas.h>
namespace migraph {
namespace miopen {
using rocblas_handle_ptr = MIGRAPH_MANAGE_PTR(rocblas_handle, rocblas_destroy_handle);
rocblas_handle_ptr create_rocblas_handle_ptr();
} // namespace miopen
} // namespace migraph
#endif
#include <migraph/miopen/rocblas.hpp>
namespace migraph {
namespace miopen {
rocblas_handle_ptr create_rocblas_handle_ptr()
{
rocblas_handle handle;
rocblas_create_handle(&handle);
return rocblas_handle_ptr{handle};
}
} // namespace miopen
} // namespace migraph
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment