h{hSSKJr SSKrSSKJr SSKJr S Sjr\RRR\\RRR\\RRR\\RRR\"\SS9\RRR\\RRR\"\SS9\RRR \"\SS9\RRR"\"\SS90rg) )partialN)SparseSemiStructuredTensor)semi_structured_sparsify_likec$^^^ Sm UHn[U[5(dMUm M T ceUU U4Sjn[U5VVs/sH upuU"Xu5PM nnnT RT RT"UV s/sH&n [U [5(a U R OU PM( sn 6T R T"UV s/sH&n [U [5(a U ROU PM( sn 6T RT R5$s snnfs sn fs sn f)am adds pointwise op support for semi-structured tensors. Assumes that at least one of the arguments in arg is a SparseSemiStructuredTensor. The last instance of a SparseSemiStructuredTensor is used as the reference mask to sparsify the others tensors passed in args. sparsify_like_args_list is used to specify which arguments to sparsify like the reference tensor. Nc>[U[R5(Ga9[U[5(d`UT;a[ UT5nU$[ STR STRS[T5S[T5SUS[U53 5eURbyTRblURR5TRR5:wd6URR5TRR5:wa>[ STR STRS[T5S[T5S3 5eU$)Nz Operation .z on z requires all operands to be z, but operand z is a z with the same sparsity pattern) isinstancetorchTensorrr ValueError __module____name__typecompressed_swizzled_bitmaskdata_ptrstride)itensorfuncreference_sparse_tensorsparsify_like_args_lists a/home/james-whalen/.local/lib/python3.13/site-packages/torchao/sparsity/training/pointwise_ops.py handle_arg-_semi_sparse_pointwise_op..handle_argse fell + +f&@AA//: 7F2 +%$T__$5Qt}}oT$OfJgIhi"#:;.JJR99BBD.JJSSUV99@@B.JJQQST%$T__$5Qt}}oT$OfJgIhiF 7899XZ ) r r enumerate __class__shapepackedmetapacked_tmeta_tr) rtypesargskwargsrrrr args_updatedxrs ` ` @r_semi_sparse_pointwise_opr(s # f8 9 9&, # # .. .Br9s  3L@BDR IINN2 IINN2 IINN2 IINN! &  IINN1 IINN  '!4# IINN  '!6# IINN%%w! $(%'#r