diff options
Diffstat (limited to 'src/amd/common')
| -rw-r--r-- | src/amd/common/ac_llvm_util.c | 31 | ||||
| -rw-r--r-- | src/amd/common/ac_llvm_util.h | 3 |
2 files changed, 34 insertions, 0 deletions
diff --git a/src/amd/common/ac_llvm_util.c b/src/amd/common/ac_llvm_util.c index b88c4e4979f..3530bf088be 100644 --- a/src/amd/common/ac_llvm_util.c +++ b/src/amd/common/ac_llvm_util.c @@ -24,10 +24,12 @@ */ /* based on pieces from si_pipe.c and radeon_llvm_emit.c */ #include "ac_llvm_util.h" +#include "ac_llvm_build.h" #include "util/bitscan.h" #include <llvm-c/Core.h> #include <llvm-c/Support.h> #include "c11/threads.h" +#include "util/u_math.h" #include <assert.h> #include <stdio.h> @@ -207,3 +209,32 @@ ac_llvm_add_target_dep_function_attr(LLVMValueRef F, snprintf(str, sizeof(str), "%i", value); LLVMAddTargetDependentFunctionAttr(F, name, str); } + +unsigned +ac_count_scratch_private_memory(LLVMValueRef function) +{ + unsigned private_mem_vgprs = 0; + + /* Process all LLVM instructions. */ + LLVMBasicBlockRef bb = LLVMGetFirstBasicBlock(function); + while (bb) { + LLVMValueRef next = LLVMGetFirstInstruction(bb); + + while (next) { + LLVMValueRef inst = next; + next = LLVMGetNextInstruction(next); + + if (LLVMGetInstructionOpcode(inst) != LLVMAlloca) + continue; + + LLVMTypeRef type = LLVMGetElementType(LLVMTypeOf(inst)); + /* No idea why LLVM aligns allocas to 4 elements. */ + unsigned alignment = LLVMGetAlignment(inst); + unsigned dw_size = align(ac_get_type_size(type) / 4, alignment); + private_mem_vgprs += dw_size; + } + bb = LLVMGetNextBasicBlock(bb); + } + + return private_mem_vgprs; +} diff --git a/src/amd/common/ac_llvm_util.h b/src/amd/common/ac_llvm_util.h index 3cf385a33ed..5329bb1b702 100644 --- a/src/amd/common/ac_llvm_util.h +++ b/src/amd/common/ac_llvm_util.h @@ -105,6 +105,9 @@ ac_get_store_intr_attribs(bool writeonly_memory) AC_FUNC_ATTR_WRITEONLY; } +unsigned +ac_count_scratch_private_memory(LLVMValueRef function); + #ifdef __cplusplus } #endif |