Home
last modified time | relevance | path

Searched refs:SparseSemiStructuredTensor (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/torch/sparse/
D_semi_structured_ops.py37 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)
Dsemi_structured.py37 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):
D__init__.py12 SparseSemiStructuredTensor,
/external/pytorch/test/
Dtest_sparse_semi_structured.py11 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 …]
/external/pytorch/benchmarks/sparse/
Dbenchmark_semi_structured_sparsity.py10 from torch.sparse import SparseSemiStructuredTensor, to_sparse_semi_structured
53 SparseSemiStructuredTensor._FORCE_CUTLASS = backend == "cutlass"
/external/pytorch/test/inductor/
Dtest_cutlass_backend.py18 from torch.sparse import SparseSemiStructuredTensor, to_sparse_semi_structured
609 SparseSemiStructuredTensor._FORCE_CUTLASS = True
/external/pytorch/docs/source/
Dsparse.rst267 …SparseSemiStructuredTensor(shape=torch.Size([128, 128]), transposed=False, values=tensor([[1., 1.,…