Add dynamic shared memory allocation #144
Conversation
|
This PR looks fine but keep in mind my remarks concerning computing that shared memory pool size statically using PE instead of a bespoke pass. Also why merge this into master considering the plugin system isn't merged there currently and probably won't be for a while ? |
|
Sorry, I should have mentioned that up there, this is independent of the plugin and programming model stuff. It just adds a parameter to allocate a given amount of dynamic shared memory upon launch. There's a corresponding PR that updates the runtime accordingly. We will need this regardless of how we end up computing the size, and this is also useful as just a standalone thing, so we thought it would be best to just get this into main right away to minimize divergence. |
michael-kenzel
force-pushed
the
mikey/smem
branch
from
80e537d to
05587c5
Compare
michael-kenzel
force-pushed
the
mikey/smem
branch
3 times, most recently
from
0509bd9 to
00bb7bd
Compare
michael-kenzel
force-pushed
the
mikey/smem
branch
2 times, most recently
from
c02028e to
bb992d8
Compare
| @@ -76,4 +76,8 @@ llvm::Value* AMDGPUCodeGen::emit_reserve(llvm::IRBuilder<>& irbuilder, const Con | |||
| return emit_reserve_shared(irbuilder, continuation, true); | |||
| } | |||
|
|
|||
| llvm::Value* AMDGPUCodeGen::emit_local_memory(llvm::IRBuilder<>& irbuilder, const Continuation* continuation) { | |||
| return emit_local_memory_base_ptr(irbuilder, continuation); | |||
| } | |||
There was a problem hiding this comment.
why is this identical function copy/pasted here and in nvvm ? since there are no alternative implementations there's no reason to do this, just have the default impl in llvm.cpp be this one
There was a problem hiding this comment.
I just did it the same way it was done for reserve_shared(). It's not valid to use this function unless one of the GPU backends is being used. So the base implementation in llvm.cpp emits an error, and the GPU backends override that with a call to the actual implementation.
michael-kenzel
force-pushed
the
mikey/smem
branch
from
bb992d8 to
ff2aaf6
Compare
This adds a parameter to allocate a given amount of dynamic shared memory upon kernel launch.
corresponding runtime changes: AnyDSL/runtime#41