Searched refs:SparseSemiStructuredTensor (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | _semi_structured_ops.py | 37 assert isinstance(A, torch.sparse.SparseSemiStructuredTensor)
50 assert isinstance(A, torch.sparse.SparseSemiStructuredTensor)
64 assert isinstance(self, torch.sparse.SparseSemiStructuredTensor)
114 if isinstance(A, torch.sparse.SparseSemiStructuredTensor):
121 assert isinstance(B_t, torch.sparse.SparseSemiStructuredTensor)
139 if isinstance(A, torch.sparse.SparseSemiStructuredTensor):
144 assert isinstance(B_t, torch.sparse.SparseSemiStructuredTensor)
|
| H A D | semi_structured.py | 37 class SparseSemiStructuredTensor(torch.Tensor): class
319 ) -> SparseSemiStructuredTensor:
377 if SparseSemiStructuredTensor._FORCE_CUTLASS
384 class SparseSemiStructuredTensorCUTLASS(SparseSemiStructuredTensor):
500 if isinstance(B, SparseSemiStructuredTensor):
523 class SparseSemiStructuredTensorCUSPARSELT(SparseSemiStructuredTensor):
554 fuse_transpose_cusparselt=SparseSemiStructuredTensor._FUSE_TRANSPOSE,
555 alg_id_cusparselt=SparseSemiStructuredTensor._DEFAULT_ALG_ID,
616 if isinstance(B, SparseSemiStructuredTensor):
|
| H A D | __init__.py | 12 SparseSemiStructuredTensor,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse_semi_structured.py | 11 SparseSemiStructuredTensor,
272 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
281 assert isinstance(A_sparse, SparseSemiStructuredTensor)
290 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
318 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
353 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
374 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
399 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
420 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
445 SparseSemiStructuredTensor._FORCE_CUTLASS = (backend == "cutlass")
[all …]
|
| /aosp_15_r20/external/pytorch/benchmarks/sparse/ |
| H A D | benchmark_semi_structured_sparsity.py | 10 from torch.sparse import SparseSemiStructuredTensor, to_sparse_semi_structured
53 SparseSemiStructuredTensor._FORCE_CUTLASS = backend == "cutlass"
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cutlass_backend.py | 18 from torch.sparse import SparseSemiStructuredTensor, to_sparse_semi_structured
609 SparseSemiStructuredTensor._FORCE_CUTLASS = True
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | sparse.rst | 267 …SparseSemiStructuredTensor(shape=torch.Size([128, 128]), transposed=False, values=tensor([[1., 1.,…
|