File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 3535
3636sys .path .insert (0 , str (Path ('..' ).resolve ()))
3737
38- autodoc_mock_imports = ['torch' , 'openequivariance.extlib' ]
38+ autodoc_mock_imports = ['torch' , 'openequivariance.extlib' , 'jinja2' ]
3939autodoc_typehints = "description"
Original file line number Diff line number Diff line change 1- sphinx-autobuild . _build --watch ../openequivariance
1+ sphinx-autobuild -M dirhtml . _build --watch ../openequivariance
Original file line number Diff line number Diff line change 11Supported Operations
2- ==============================
2+ ====================
33
44.. toctree ::
55 :maxdepth: 1
@@ -28,20 +28,6 @@ Supported Operations
2828 - ✅ (beta)
2929 - ✅ (beta)
3030
31- +----------------------------+----------+------------+
32- | Operation | CUDA | HIP |
33- +============================+==========+============+
34- | UVU | ✅ | ✅ |
35- +----------------------------+----------+------------+
36- | UVW | ✅ | ✅ |
37- +----------------------------+----------+------------+
38- | UVU + Convolution | ✅ | ✅ |
39- +----------------------------+----------+------------+
40- | UVW + Convolution | ✅ | ✅ |
41- +----------------------------+----------+------------+
42- | Symmetric Tensor Product | ✅ (beta) | ✅ (beta) |
43- +----------------------------+----------+------------+
44-
4531e3nn supports a variety of connection modes for CG tensor products. We support
4632two that are commonly used in equivariant graph neural networks:
4733"uvu" and "uvw". Our JIT compiled kernels should handle:
You can’t perform that action at this time.
0 commit comments