Commit b688bf6a authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

remove sync device and other tags

parent 1d0f8ead
...@@ -200,7 +200,6 @@ hip::copy_from_gpu ...@@ -200,7 +200,6 @@ hip::copy_from_gpu
hip::copy_to_gpu hip::copy_to_gpu
hip::hip_allocate_memory hip::hip_allocate_memory
hip::hip_copy_literal hip::hip_copy_literal
hip::sync_device
identity identity
im2col im2col
leaky_relu leaky_relu
......
...@@ -14,7 +14,6 @@ inline namespace MIGRAPHX_INLINE_NS { ...@@ -14,7 +14,6 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace gpu { namespace gpu {
MIGRAPHX_REGISTER_OP(hip_allocate) MIGRAPHX_REGISTER_OP(hip_allocate)
MIGRAPHX_REGISTER_OP(hip_sync_device)
MIGRAPHX_REGISTER_OP(hip_sync_stream) MIGRAPHX_REGISTER_OP(hip_sync_stream)
MIGRAPHX_REGISTER_OP(hip_copy_to_gpu) MIGRAPHX_REGISTER_OP(hip_copy_to_gpu)
MIGRAPHX_REGISTER_OP(hip_copy_from_gpu) MIGRAPHX_REGISTER_OP(hip_copy_from_gpu)
......
...@@ -36,12 +36,11 @@ argument get_preallocation(context& ctx, const std::string& id); ...@@ -36,12 +36,11 @@ argument get_preallocation(context& ctx, const std::string& id);
struct hip_allocate struct hip_allocate
{ {
shape s; shape s;
std::string tag{};
template <class Self, class F> template <class Self, class F>
static auto reflect(Self& self, F f) static auto reflect(Self& self, F f)
{ {
return pack(f(self.s, "shape"), f(self.tag, "tag")); return pack(f(self.s, "shape"));
} }
std::string name() const { return "hip::allocate"; } std::string name() const { return "hip::allocate"; }
...@@ -56,42 +55,8 @@ struct hip_allocate ...@@ -56,42 +55,8 @@ struct hip_allocate
} }
}; };
struct hip_sync_device
{
std::string tag{};
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return pack(f(self.tag, "tag"));
}
std::string name() const { return "hip::sync_device"; }
shape compute_shape(const std::vector<shape>& inputs) const
{
if(inputs.empty())
return {};
return inputs.front();
}
argument compute(context&, const shape&, const std::vector<argument>& args) const
{
gpu_sync();
if(args.empty())
return {};
return args.front();
}
};
struct hip_sync_stream struct hip_sync_stream
{ {
std::string tag{};
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return pack(f(self.tag, "tag"));
}
std::string name() const { return "hip::sync_stream"; } std::string name() const { return "hip::sync_stream"; }
shape compute_shape(const std::vector<shape>& inputs) const shape compute_shape(const std::vector<shape>& inputs) const
......
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