Skip to content

Commit 73379b8

Browse files
committed
Update on "[ET-VK][7/n] Slice, with lots of codegen improvements"
1. Add slice operation. Instead of using copy in LI, we implement a simple shader with offsets. 2. Improvement in codegen. - add support of optional variables - improve indent of the code, for better readability - allow user to specify tensor value generation, possible to generate sequential values for easier debugging for index operations - sample code improve test-case specification, particularly with long and optional values. Differential Revision: [D56295985](https://our.internmc.facebook.com/intern/diff/D56295985/) [ghstack-poisoned]
1 parent f6194e6 commit 73379b8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/vulkan/test/op_tests/utils/codegen_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def gen_case_name(self, inputs: List[Any], prepack: bool = False) -> str:
117117
name_str = name_str.replace("-", "n")
118118
return name_str
119119

120-
def create_input_data(self, arg: Argument, data: Any) -> str:
120+
def create_input_data(self, arg: Argument, data: Any) -> str: # noqa: C901
121121
ctype = cpp.argumenttype_type(arg.type, mutable=arg.is_write, binds=arg.name)
122122
cpp_type = ctype.cpp_type(strip_ref=True)
123123

0 commit comments

Comments
 (0)