-
Notifications
You must be signed in to change notification settings - Fork 40
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Visualizer Ver1.0 #113
Visualizer Ver1.0 #113
Conversation
Fix the casting issues between Cython and Python objects
deps/cutlass
Outdated
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We may want to update your cutlass commit. You can run git submodule update
to update it.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ok
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I have checkouted cutlass submodule to commit cc3c29a
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The PR looks great overall. I think it will be ready to be merged after addressing these minor issues. Thanks!
@@ -464,6 +833,32 @@ cdef class CyTBGraph: | |||
t = ctypes.cast(<unsigned long long>ptr, ctypes.c_void_p) | |||
return STensor(t) | |||
|
|||
property grid_dim: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I was wondering if we can unify grid_dim
and block_dim
here. For example, both of them can return a dictionary or a tuple. Let's try not to use list.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It seems like block_dim is not used in the graph shown in the paper and actually also not used in the graph visualizer generated🤔. I removed it in current version.
… in func and remove block_dim
Description of changes:
Embedded visualizer in python interface.
Simply use
graph.visualize("graph_name", "file_name")
to generate the graph.Will support the newly added output node.
Related Issues:
Linked Issues:
Issues closed by this PR: