Unverified Commit d1b5f332 authored by Charlie Lin's avatar Charlie Lin Committed by GitHub
Browse files

Initial (#1586)

Add additional documentation to explain the passes.
parent a63ee2e0
...@@ -33,7 +33,8 @@ inline namespace MIGRAPHX_INLINE_NS { ...@@ -33,7 +33,8 @@ inline namespace MIGRAPHX_INLINE_NS {
struct module; struct module;
/** /**
* Remove memory allocations. It uses graph coloring to find memory allocations that can be reused. * Remove multiple memory allocations using graph coloring to find memory allocations that can be
* reused.
*/ */
struct memory_coloring struct memory_coloring
{ {
......
...@@ -32,6 +32,9 @@ inline namespace MIGRAPHX_INLINE_NS { ...@@ -32,6 +32,9 @@ inline namespace MIGRAPHX_INLINE_NS {
struct module; struct module;
/**
* Replace `allocate` instructions with target allocations or output parameters.
*/
struct replace_allocate struct replace_allocate
{ {
allocation_model model; allocation_model model;
......
...@@ -33,6 +33,13 @@ inline namespace MIGRAPHX_INLINE_NS { ...@@ -33,6 +33,13 @@ inline namespace MIGRAPHX_INLINE_NS {
struct module; struct module;
namespace gpu { namespace gpu {
/**
* Compiler pass that makes GPU-specific instruction changes.
* * Copies to and from the device if `offload_copy` is true.
* * Maps instructions to their GPU-specific counterparts.
* * Inserts `allocate` instructions before GPU operators.
*/
struct lowering struct lowering
{ {
context* ctx; context* ctx;
......
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