Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
025894f3
Commit
025894f3
authored
Apr 18, 2025
by
PanZezhong
Browse files
issue/48/fix 将rope info的workspace_size改成私有
parent
c905fd63
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
47 additions
and
46 deletions
+47
-46
src/infiniop/ops/rope/operator.cc
src/infiniop/ops/rope/operator.cc
+3
-3
src/infiniop/ops/rope/rope.h
src/infiniop/ops/rope/rope.h
+44
-43
No files found.
src/infiniop/ops/rope/operator.cc
View file @
025894f3
...
@@ -75,7 +75,7 @@ __C infiniStatus_t infiniopGetRoPEWorkspaceSize(infiniopRoPEDescriptor_t desc,
...
@@ -75,7 +75,7 @@ __C infiniStatus_t infiniopGetRoPEWorkspaceSize(infiniopRoPEDescriptor_t desc,
size_t
*
size
)
{
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
*size = reinterpret_cast<const op::rope::NAMESPACE::Descriptor *>(desc)->workspace
_s
ize; \
*size = reinterpret_cast<const op::rope::NAMESPACE::Descriptor *>(desc)->workspace
S
ize
()
; \
return INFINI_STATUS_SUCCESS
return INFINI_STATUS_SUCCESS
switch
(
desc
->
device_type
)
{
switch
(
desc
->
device_type
)
{
...
...
src/infiniop/ops/rope/rope.h
View file @
025894f3
...
@@ -12,6 +12,7 @@
...
@@ -12,6 +12,7 @@
struct Opaque; \
struct Opaque; \
Opaque *_opaque; \
Opaque *_opaque; \
RoPEInfo _info; \
RoPEInfo _info; \
size_t _workspace_size; \
\
\
Descriptor( \
Descriptor( \
RoPEInfo info, \
RoPEInfo info, \
...
@@ -22,13 +23,13 @@
...
@@ -22,13 +23,13 @@
: InfiniopDescriptor{device_type, device_id}, \
: InfiniopDescriptor{device_type, device_id}, \
_opaque(opaque), \
_opaque(opaque), \
_info(info), \
_info(info), \
workspace_size(workspace_size_) {} \
_
workspace_size(workspace_size_) {}
\
\
\
public: \
public: \
size_t workspace_size; \
\
~Descriptor(); \
~Descriptor(); \
\
\
size_t workspaceSize() const { return _workspace_size; } \
\
static infiniStatus_t create( \
static infiniStatus_t create( \
infiniopHandle_t handle, \
infiniopHandle_t handle, \
Descriptor **desc_ptr, \
Descriptor **desc_ptr, \
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment