Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
M
memory-faucet
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
数学の武士
memory-faucet
Commits
4bc91d97
Commit
4bc91d97
authored
7 months ago
by
数学の武士
Browse files
Options
Downloads
Patches
Plain Diff
Change memory_pipeline* name to memory_pipe*
parent
0c262ac2
Branches
Branches containing commit
No related tags found
No related merge requests found
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
Lib/memory-holder.cpp
+24
-24
24 additions, 24 deletions
Lib/memory-holder.cpp
Lib/memory-holder.h
+19
-19
19 additions, 19 deletions
Lib/memory-holder.h
with
43 additions
and
43 deletions
Lib/memory-holder.cpp
+
24
−
24
View file @
4bc91d97
...
@@ -151,7 +151,7 @@ template class buffer<MemType::CPU>;
...
@@ -151,7 +151,7 @@ template class buffer<MemType::CPU>;
template
class
buffer
<
MemType
::
GPU
>;
template
class
buffer
<
MemType
::
GPU
>;
#endif
#endif
memory_pip
lin
e_base
::
memory_pip
lin
e_base
()
memory_pipe_base
::
memory_pipe_base
()
{
{
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
gpu_buff
=
std
::
vector
<
buffer
<
MemType
::
GPU
>
>
();
gpu_buff
=
std
::
vector
<
buffer
<
MemType
::
GPU
>
>
();
...
@@ -159,7 +159,7 @@ memory_pipline_base::memory_pipline_base()
...
@@ -159,7 +159,7 @@ memory_pipline_base::memory_pipline_base()
cpu_buff
=
std
::
vector
<
buffer
<
MemType
::
CPU
>
>
();
cpu_buff
=
std
::
vector
<
buffer
<
MemType
::
CPU
>
>
();
}
}
memory_pip
lin
e_base
::~
memory_pip
lin
e_base
()
memory_pipe_base
::~
memory_pipe_base
()
{
{
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
gpu_buff
.
clear
();
gpu_buff
.
clear
();
...
@@ -168,28 +168,28 @@ memory_pipline_base::~memory_pipline_base()
...
@@ -168,28 +168,28 @@ memory_pipline_base::~memory_pipline_base()
}
}
template
<
MemType
mem
>
template
<
MemType
mem
>
std
::
vector
<
buffer
<
mem
>
>&
memory_pip
lin
e_base
::
get_memtyped_vector
()
std
::
vector
<
buffer
<
mem
>
>&
memory_pipe_base
::
get_memtyped_vector
()
{
{
return
cpu_buff
;
return
cpu_buff
;
}
}
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
template
<
>
template
<
>
std
::
vector
<
buffer
<
MemType
::
GPU
>
>&
memory_pip
lin
e_base
::
get_memtyped_vector
()
std
::
vector
<
buffer
<
MemType
::
GPU
>
>&
memory_pipe_base
::
get_memtyped_vector
()
{
{
return
gpu_buff
;
return
gpu_buff
;
}
}
#endif
#endif
template
<
MemType
mem
>
template
<
MemType
mem
>
void
memory_pip
lin
e_base
::
set_available
(
const
int
id
)
void
memory_pipe_base
::
set_available
(
const
int
id
)
{
{
get_memtyped_vector
<
mem
>
()[
id
].
set_status
(
true
);
get_memtyped_vector
<
mem
>
()[
id
].
set_status
(
true
);
}
}
template
void
memory_pip
lin
e_base
::
set_available
<
MemType
::
CPU
>(
const
int
id
);
template
void
memory_pipe_base
::
set_available
<
MemType
::
CPU
>(
const
int
id
);
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
template
void
memory_pip
lin
e_base
::
set_available
<
MemType
::
GPU
>(
const
int
id
);
template
void
memory_pipe_base
::
set_available
<
MemType
::
GPU
>(
const
int
id
);
#endif
#endif
template
<
MemType
mem
>
template
<
MemType
mem
>
...
@@ -260,7 +260,7 @@ template typename std::vector<buffer<MemType::GPU>>::iterator get_upper_bound<Me
...
@@ -260,7 +260,7 @@ template typename std::vector<buffer<MemType::GPU>>::iterator get_upper_bound<Me
template
<
buf_choose_policy
choose_type
>
template
<
buf_choose_policy
choose_type
>
template
<
MemType
mem
>
template
<
MemType
mem
>
int
memory_pip
lin
e
<
choose_type
>::
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
)
int
memory_pipe
<
choose_type
>::
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
)
{
{
std
::
vector
<
buffer
<
mem
>
>&
buff_vec
=
get_memtyped_vector
<
mem
>
();
std
::
vector
<
buffer
<
mem
>
>&
buff_vec
=
get_memtyped_vector
<
mem
>
();
const
int
allocated_buffer_n
=
buff_vec
.
size
();
const
int
allocated_buffer_n
=
buff_vec
.
size
();
...
@@ -282,13 +282,13 @@ int memory_pipline<choose_type>::get_buffer(const size_t required_size, void *&
...
@@ -282,13 +282,13 @@ int memory_pipline<choose_type>::get_buffer(const size_t required_size, void *&
return
id
;
return
id
;
}
}
template
int
memory_pip
lin
e
<
buf_choose_policy
::
naiv
>
::
get_buffer
<
MemType
::
CPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
template
int
memory_pipe
<
buf_choose_policy
::
naiv
>
::
get_buffer
<
MemType
::
CPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
template
int
memory_pip
lin
e
<
buf_choose_policy
::
naiv
>
::
get_buffer
<
MemType
::
GPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
template
int
memory_pipe
<
buf_choose_policy
::
naiv
>
::
get_buffer
<
MemType
::
GPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
#endif
#endif
template
<
MemType
mem
>
template
<
MemType
mem
>
int
memory_pip
lin
e
<
buf_choose_policy
::
sorted_vec
>::
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
)
int
memory_pipe
<
buf_choose_policy
::
sorted_vec
>::
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
)
{
{
std
::
vector
<
buffer
<
mem
>
>&
buff_vec
=
get_memtyped_vector
<
mem
>
();
std
::
vector
<
buffer
<
mem
>
>&
buff_vec
=
get_memtyped_vector
<
mem
>
();
const
int
allocated_buffer_n
=
buff_vec
.
size
();
const
int
allocated_buffer_n
=
buff_vec
.
size
();
...
@@ -314,13 +314,13 @@ int memory_pipline<buf_choose_policy::sorted_vec>::get_buffer(const size_t requi
...
@@ -314,13 +314,13 @@ int memory_pipline<buf_choose_policy::sorted_vec>::get_buffer(const size_t requi
return
id
;
return
id
;
}
}
template
int
memory_pip
lin
e
<
buf_choose_policy
::
sorted_vec
>
::
get_buffer
<
MemType
::
CPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
template
int
memory_pipe
<
buf_choose_policy
::
sorted_vec
>
::
get_buffer
<
MemType
::
CPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
template
int
memory_pip
lin
e
<
buf_choose_policy
::
sorted_vec
>
::
get_buffer
<
MemType
::
GPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
template
int
memory_pipe
<
buf_choose_policy
::
sorted_vec
>
::
get_buffer
<
MemType
::
GPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
#endif
#endif
template
<
MemType
mem
>
template
<
MemType
mem
>
int
memory_pip
lin
e
<
buf_choose_policy
::
find_best_unsorted
>::
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
)
int
memory_pipe
<
buf_choose_policy
::
find_best_unsorted
>::
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
)
{
{
std
::
vector
<
buffer
<
mem
>
>&
buff_vec
=
get_memtyped_vector
<
mem
>
();
std
::
vector
<
buffer
<
mem
>
>&
buff_vec
=
get_memtyped_vector
<
mem
>
();
typename
std
::
vector
<
buffer
<
mem
>>::
iterator
available_buf_it
;
typename
std
::
vector
<
buffer
<
mem
>>::
iterator
available_buf_it
;
...
@@ -341,29 +341,29 @@ int memory_pipline<buf_choose_policy::find_best_unsorted>::get_buffer(const size
...
@@ -341,29 +341,29 @@ int memory_pipline<buf_choose_policy::find_best_unsorted>::get_buffer(const size
}
}
}
}
template
int
memory_pip
lin
e
<
buf_choose_policy
::
find_best_unsorted
>
::
get_buffer
<
MemType
::
CPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
template
int
memory_pipe
<
buf_choose_policy
::
find_best_unsorted
>
::
get_buffer
<
MemType
::
CPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
template
int
memory_pip
lin
e
<
buf_choose_policy
::
find_best_unsorted
>
::
get_buffer
<
MemType
::
GPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
template
int
memory_pipe
<
buf_choose_policy
::
find_best_unsorted
>
::
get_buffer
<
MemType
::
GPU
>
(
const
size_t
required_size
,
void
*&
ptr
);
#endif
#endif
template
<
>
template
<
>
memory_pip
lin
e
<
buf_choose_policy
::
naiv
>&
memory_faucet
::
get_faucet
()
memory_pipe
<
buf_choose_policy
::
naiv
>&
memory_faucet
::
get_faucet
()
{
{
static
memory_pip
lin
e
<
buf_choose_policy
::
naiv
>
mem_pipe_naiv
;
static
memory_pipe
<
buf_choose_policy
::
naiv
>
mem_pipe_naiv
;
return
mem_pipe_naiv
;
return
mem_pipe_naiv
;
}
}
template
<
>
template
<
>
memory_pip
lin
e
<
buf_choose_policy
::
sorted_vec
>&
memory_faucet
::
get_faucet
()
memory_pipe
<
buf_choose_policy
::
sorted_vec
>&
memory_faucet
::
get_faucet
()
{
{
static
memory_pip
lin
e
<
buf_choose_policy
::
sorted_vec
>
mem_pipe_sorted
;
static
memory_pipe
<
buf_choose_policy
::
sorted_vec
>
mem_pipe_sorted
;
return
mem_pipe_sorted
;
return
mem_pipe_sorted
;
}
}
template
<
>
template
<
>
memory_pip
lin
e
<
buf_choose_policy
::
find_best_unsorted
>&
memory_faucet
::
get_faucet
()
memory_pipe
<
buf_choose_policy
::
find_best_unsorted
>&
memory_faucet
::
get_faucet
()
{
{
static
memory_pip
lin
e
<
buf_choose_policy
::
find_best_unsorted
>
mem_pipe_unsorted
;
static
memory_pipe
<
buf_choose_policy
::
find_best_unsorted
>
mem_pipe_unsorted
;
return
mem_pipe_unsorted
;
return
mem_pipe_unsorted
;
}
}
...
@@ -371,7 +371,7 @@ memory_pipline<buf_choose_policy::find_best_unsorted>& memory_faucet::get_faucet
...
@@ -371,7 +371,7 @@ memory_pipline<buf_choose_policy::find_best_unsorted>& memory_faucet::get_faucet
template
<
MemType
mem
,
buf_choose_policy
choose_type
>
template
<
MemType
mem
,
buf_choose_policy
choose_type
>
memBuf
<
mem
,
choose_type
>::
memBuf
(
const
size_t
required_size
)
memBuf
<
mem
,
choose_type
>::
memBuf
(
const
size_t
required_size
)
{
{
memory_pip
lin
e
<
choose_type
>&
mem_pipe
=
memory_faucet
::
get_faucet
<
choose_type
>
();
memory_pipe
<
choose_type
>&
mem_pipe
=
memory_faucet
::
get_faucet
<
choose_type
>
();
id
=
mem_pipe
.
template
get_buffer
<
mem
>(
required_size
,
buf
);
id
=
mem_pipe
.
template
get_buffer
<
mem
>(
required_size
,
buf
);
size
=
required_size
;
size
=
required_size
;
}
}
...
@@ -379,7 +379,7 @@ memBuf<mem, choose_type>::memBuf(const size_t required_size)
...
@@ -379,7 +379,7 @@ memBuf<mem, choose_type>::memBuf(const size_t required_size)
template
<
MemType
mem
,
buf_choose_policy
choose_type
>
template
<
MemType
mem
,
buf_choose_policy
choose_type
>
memBuf
<
mem
,
choose_type
>::~
memBuf
()
memBuf
<
mem
,
choose_type
>::~
memBuf
()
{
{
memory_pip
lin
e
<
choose_type
>&
mem_pipe
=
memory_faucet
::
get_faucet
<
choose_type
>
();
memory_pipe
<
choose_type
>&
mem_pipe
=
memory_faucet
::
get_faucet
<
choose_type
>
();
mem_pipe
.
template
set_available
<
mem
>(
id
);
mem_pipe
.
template
set_available
<
mem
>(
id
);
}
}
...
...
This diff is collapsed.
Click to expand it.
Lib/memory-holder.h
+
19
−
19
View file @
4bc91d97
...
@@ -41,7 +41,7 @@ public:
...
@@ -41,7 +41,7 @@ public:
void
set_id
(
const
int
id
);
void
set_id
(
const
int
id
);
};
};
class
memory_pip
lin
e_base
class
memory_pipe_base
{
{
public:
public:
#ifdef INCLUDE_CUDA
#ifdef INCLUDE_CUDA
...
@@ -49,8 +49,8 @@ public:
...
@@ -49,8 +49,8 @@ public:
#endif
#endif
std
::
vector
<
buffer
<
MemType
::
CPU
>
>
cpu_buff
;
std
::
vector
<
buffer
<
MemType
::
CPU
>
>
cpu_buff
;
memory_pip
lin
e_base
();
memory_pipe_base
();
~
memory_pip
lin
e_base
();
~
memory_pipe_base
();
template
<
MemType
mem
>
template
<
MemType
mem
>
void
set_available
(
const
int
id
);
void
set_available
(
const
int
id
);
...
@@ -63,48 +63,48 @@ public:
...
@@ -63,48 +63,48 @@ public:
};
};
template
<
buf_choose_policy
choose_type
=
buf_choose_policy
::
naiv
>
template
<
buf_choose_policy
choose_type
=
buf_choose_policy
::
naiv
>
class
memory_pip
lin
e
:
public
memory_pip
lin
e_base
class
memory_pipe
:
public
memory_pipe_base
{
{
// private:
// private:
// #ifdef INCLUDE_CUDA
// #ifdef INCLUDE_CUDA
// using memory_pip
lin
e_base::gpu_buff;
// using memory_pipe_base::gpu_buff;
// #endif
// #endif
// using memory_pip
lin
e_base::cpu_buff;
// using memory_pipe_base::cpu_buff;
public:
public:
memory_pip
lin
e
(
/* args */
)
:
memory_pip
lin
e_base
()
{}
memory_pipe
(
/* args */
)
:
memory_pipe_base
()
{}
~
memory_pip
lin
e
()
=
default
;
~
memory_pipe
()
=
default
;
template
<
MemType
mem
>
template
<
MemType
mem
>
int
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
);
int
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
);
};
};
template
<
>
template
<
>
class
memory_pip
lin
e
<
buf_choose_policy
::
sorted_vec
>:
public
memory_pip
lin
e_base
class
memory_pipe
<
buf_choose_policy
::
sorted_vec
>:
public
memory_pipe_base
{
{
// private:
// private:
// #ifdef INCLUDE_CUDA
// #ifdef INCLUDE_CUDA
// using memory_pip
lin
e_base::gpu_buff;
// using memory_pipe_base::gpu_buff;
// #endif
// #endif
// using memory_pip
lin
e_base::cpu_buff;
// using memory_pipe_base::cpu_buff;
public:
public:
memory_pip
lin
e
(
/* args */
)
:
memory_pip
lin
e_base
()
{}
memory_pipe
(
/* args */
)
:
memory_pipe_base
()
{}
~
memory_pip
lin
e
()
=
default
;
~
memory_pipe
()
=
default
;
template
<
MemType
mem
>
template
<
MemType
mem
>
int
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
);
int
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
);
};
};
template
<
>
template
<
>
class
memory_pip
lin
e
<
buf_choose_policy
::
find_best_unsorted
>:
public
memory_pip
lin
e_base
class
memory_pipe
<
buf_choose_policy
::
find_best_unsorted
>:
public
memory_pipe_base
{
{
// private:
// private:
// #ifdef INCLUDE_CUDA
// #ifdef INCLUDE_CUDA
// using memory_pip
lin
e_base::gpu_buff;
// using memory_pipe_base::gpu_buff;
// #endif
// #endif
// using memory_pip
lin
e_base::cpu_buff;
// using memory_pipe_base::cpu_buff;
public:
public:
memory_pip
lin
e
(
/* args */
)
:
memory_pip
lin
e_base
()
{}
memory_pipe
(
/* args */
)
:
memory_pipe_base
()
{}
~
memory_pip
lin
e
()
=
default
;
~
memory_pipe
()
=
default
;
template
<
MemType
mem
>
template
<
MemType
mem
>
int
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
);
int
get_buffer
(
const
size_t
required_size
,
void
*&
ptr
);
...
@@ -120,7 +120,7 @@ private:
...
@@ -120,7 +120,7 @@ private:
public:
public:
template
<
buf_choose_policy
choose_type
=
buf_choose_policy
::
naiv
>
template
<
buf_choose_policy
choose_type
=
buf_choose_policy
::
naiv
>
static
memory_pip
lin
e
<
choose_type
>&
get_faucet
();
static
memory_pipe
<
choose_type
>&
get_faucet
();
};
};
template
<
MemType
mem
,
buf_choose_policy
choose_type
=
buf_choose_policy
::
naiv
>
template
<
MemType
mem
,
buf_choose_policy
choose_type
=
buf_choose_policy
::
naiv
>
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment