diff options
author | Jan Vesely <jano.vesely@gmail.com> | 2017-10-01 03:30:16 -0400 |
---|---|---|
committer | Jan Vesely <jan.vesely@rutgers.edu> | 2017-10-25 12:01:03 -0400 |
commit | 477b10671f318c2295524e40f5947089454573f7 (patch) | |
tree | 6385131d8d932618abe23ee46dffe434b967d5ea /generated_tests | |
parent | cefc2cf804ebee1047655731458de1bf1dafb953 (diff) |
cl: Implement generating 2input 2output functions
Signed-off-by: Jan Vesely <jano.vesely@gmail.com>
Diffstat (limited to 'generated_tests')
-rw-r--r-- | generated_tests/genclbuiltins.py | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/generated_tests/genclbuiltins.py b/generated_tests/genclbuiltins.py index 338914cb9..25f2f7a9a 100644 --- a/generated_tests/genclbuiltins.py +++ b/generated_tests/genclbuiltins.py @@ -172,10 +172,10 @@ def gen_kernel_1_arg(f, fnName, inType, outTypes, loc = 'private'): # 2 argument kernel with input types that match their vector size -def gen_kernel_2_arg_same_size(f, fnName, inTypes, outTypes): +def gen_kernel_2_arg_same_size(f, fnName, inTypes, outTypes, loc = 'private'): for vecSize in ALL_WIDTHS: gen_kernel(f, fnName, inTypes, outTypes, [vecSize, vecSize], - '') + '', loc) # 2 argument kernel with 1 vector and one scalar input argument @@ -238,8 +238,16 @@ def generate_kernels(f, dataType, fnName, fnDef): return if (len(argTypes) == 4): - gen_kernel_3_arg_same_type(f, fnName, - [argTypes[1], argTypes[2], argTypes[3]], [argTypes[0]]) + if (getNumOutArgs(fnDef) == 2): + gen_kernel_2_arg_same_size(f, fnName, [argTypes[2], argTypes[3]], + [argTypes[0], argTypes[1]], 'private') + gen_kernel_2_arg_same_size(f, fnName, [argTypes[2], argTypes[3]], + [argTypes[0], argTypes[1]], 'local') + gen_kernel_2_arg_same_size(f, fnName, [argTypes[2], argTypes[3]], + [argTypes[0], argTypes[1]], 'global') + else: + gen_kernel_3_arg_same_type(f, fnName, + [argTypes[1], argTypes[2], argTypes[3]], [argTypes[0]]) if (fnDef['function_type'] is 'tss'): gen_kernel_3_arg_mixed_size_tss(f, fnName, [argTypes[1], argTypes[2], argTypes[3]], [argTypes[0]]) |