ac: rework ac_llvm_extract_elem()

Simplifies the logic a little and asserts index is 0.

Suggested-by: Nicolai Hähnle <nhaehnle@gmail.com>
Reviewed-by: Marek Olšák <marek.olsak@amd.com>

diff --git a/src/amd/common/ac_llvm_build.c b/src/amd/common/ac_llvm_build.c
index 0ea5e7f4caacf90d98b7518b25b93aaa822c316b..8a3a2abf17b525dc5cd8efda2006f2f424963d2d 100644 (file)
--- a/src/amd/common/ac_llvm_build.c
+++ b/src/amd/common/ac_llvm_build.c
@@ -114,10 +114,10 @@ ac_llvm_extract_elem(struct ac_llvm_context *ac,
                     LLVMValueRef value,
                     int index)
 {
-       int count = ac_get_llvm_num_components(value);
-
-       if (count == 1)
+       if (LLVMGetTypeKind(LLVMTypeOf(value)) != LLVMVectorTypeKind) {
+               assert(index == 0);
                return value;
+       }
 
        return LLVMBuildExtractElement(ac->builder, value,
                                       LLVMConstInt(ac->i32, index, false), "");