Commit 7569675e authored by Graylatzhou's avatar Graylatzhou
Browse files

issue/183 规范格式

parent 734c00a6
......@@ -25,7 +25,7 @@ DECLARE_INFINIOP_TEST(mul)
{ \
REGISTER_INFINIOP_TEST(gemm) \
REGISTER_INFINIOP_TEST(random_sample) \
REGISTER_INFINIOP_TEST(mul) \
REGISTER_INFINIOP_TEST(mul) \
}
namespace infiniop_test {
......
......@@ -40,10 +40,10 @@ std::shared_ptr<infiniop_test::Result> Test::run(
auto b = _attributes->b->to(device, device_id);
auto c = _attributes->c->to(device, device_id);
CHECK_OR(infiniopCreateMulDescriptor(handle, &op_desc,
c->desc(),
a->desc(),
b->desc()),
return TEST_FAILED(OP_CREATION_FAILED, "Failed to create op descriptor."));
c->desc(),
a->desc(),
b->desc()),
return TEST_FAILED(OP_CREATION_FAILED, "Failed to create op descriptor."));
size_t workspace_size;
CHECK_OR(infiniopGetMulWorkspaceSize(op_desc, &workspace_size),
return TEST_FAILED(OP_CREATION_FAILED, "Failed to get workspace size."));
......@@ -51,12 +51,12 @@ std::shared_ptr<infiniop_test::Result> Test::run(
CHECK_OR(infinirtMalloc(&workspace, workspace_size),
return TEST_FAILED(OP_CREATION_FAILED, "Failed to allocate workspace."));
CHECK_OR(infiniopMul(op_desc, workspace, workspace_size,
c->data(),
a->data(),
b->data(),
nullptr),
c->data(),
a->data(),
b->data(),
nullptr),
return TEST_FAILED(OP_EXECUTION_FAILED, "Failed during execution."));
try {
allClose(c, _attributes->ans, _rtol, _atol);
} catch (const std::exception &e) {
......
......@@ -27,7 +27,7 @@ infiniStatus_t Descriptor::create(
CREATE_ELEMENTWISE_CPU_DESCRIPTOR(handle, dtype, out_desc, input_desc_vec);
return INFINI_STATUS_SUCCESS;
}
}
infiniStatus_t Descriptor::calculate(
void *workspace,
......@@ -36,16 +36,16 @@ infiniStatus_t Descriptor::calculate(
std::vector<const void *> inputs,
void *stream) const {
switch(_dtype) {
case INFINI_DTYPE_F16:
return _device_info->calculate<MulOp, fp16_t>(_info, output, inputs, stream);
case INFINI_DTYPE_F32:
return _device_info->calculate<MulOp, float>(_info, output, inputs, stream);
case INFINI_DTYPE_F64:
return _device_info->calculate<MulOp, double>(_info, output, inputs, stream);
default:
return INFINI_STATUS_BAD_TENSOR_DTYPE;
switch (_dtype) {
case INFINI_DTYPE_F16:
return _device_info->calculate<MulOp, fp16_t>(_info, output, inputs, stream);
case INFINI_DTYPE_F32:
return _device_info->calculate<MulOp, float>(_info, output, inputs, stream);
case INFINI_DTYPE_F64:
return _device_info->calculate<MulOp, double>(_info, output, inputs, stream);
default:
return INFINI_STATUS_BAD_TENSOR_DTYPE;
}
return INFINI_STATUS_SUCCESS;
}
}
} // namespace op::mul::cpu
......@@ -10,7 +10,6 @@
#include "cuda/mul_cuda.cuh"
#endif
__C infiniStatus_t infiniopCreateMulDescriptor(
infiniopHandle_t handle,
infiniopMulDescriptor_t *desc_ptr,
......
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