File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed
utils/bazel/llvm-project-overlay/mlir Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -484,7 +484,8 @@ def gentbl_sharded_ops(
484
484
test = False ,
485
485
includes = [],
486
486
strip_include_prefix = None ,
487
- deps = []):
487
+ deps = [],
488
+ ** kwargs ):
488
489
"""Generate sharded op declarations and definitions.
489
490
490
491
This special build rule shards op definitions in a TableGen file and generates multiple copies
@@ -524,6 +525,7 @@ def gentbl_sharded_ops(
524
525
td_file = td_file ,
525
526
test = test ,
526
527
deps = deps ,
528
+ ** kwargs ,
527
529
)
528
530
all_files = [hdr_out , src_out ]
529
531
for i in range (0 , shard_count ):
@@ -535,9 +537,14 @@ def gentbl_sharded_ops(
535
537
out = out_file ,
536
538
sharder = sharder ,
537
539
src_file = src_file ,
540
+ ** kwargs ,
538
541
)
539
542
all_files .append (out_file )
540
- native .filegroup (name = name , srcs = all_files )
543
+ native .filegroup (
544
+ name = name ,
545
+ srcs = all_files ,
546
+ ** kwargs ,
547
+ )
541
548
542
549
def gentbl_sharded_op_defs (name , source_file , shard_count ):
543
550
"""Generates multiple copies of a source file that includes sharded op definitions.
You can’t perform that action at this time.
0 commit comments