• Harsh Patel's avatar
    Optimizes `yaw_quat` implementation (#2247) · 3b8da1d3
    Harsh Patel authored
    # Description
    
    Added optimizations to the yaw_quat function and added a function to
    test_math to make sure it works as intended.
    
    **BENCHMARKS:**
    
    Device: `cpu`
    
    ```
    Time for yaw_quat: <torch.utils.benchmark.utils.common.Measurement object at 0x7a342d1f8640>
    math_utils.yaw_quat(q_rand)  `511.79 us`
    1 measurement, 1000 runs , 1 thread
    
    Time for iter_yaw_quat: <torch.utils.benchmark.utils.common.Measurement object at 0x7a342d1f9720>
    iter_yaw_quat(q_rand)  `2.18 ms`
    1 measurement, 1000 runs , 1 thread
    
    Time for iter_old_yaw_quat: <torch.utils.benchmark.utils.common.Measurement object at 0x7a342d1f8640>
    iter_old_yaw_quat(q_rand)  `2.36 ms`
    1 measurement, 1000 runs , 1 thread
    ```
    
    ---
    
    Device: `cuda:0`
    
    ```
    Time for yaw_quat: <torch.utils.benchmark.utils.common.Measurement object at 0x7a342d1f8730>
    math_utils.yaw_quat(q_rand)  `246.16 us`
    1 measurement, 1000 runs , 1 thread
    
    Time for iter_yaw_quat: <torch.utils.benchmark.utils.common.Measurement object at 0x7a342d1f8100>
    iter_yaw_quat(q_rand)  `2.97 ms`
    1 measurement, 1000 runs , 1 thread
    
    Time for iter_old_yaw_quat: <torch.utils.benchmark.utils.common.Measurement object at 0x7a342d1f8730>
    iter_old_yaw_quat(q_rand)  `3.46 ms`
    1 measurement, 1000 runs , 1 thread
    ```
    
    ## Type of change
    
    - New feature (non-breaking change which adds functionality)
    
    ## Checklist
    
    - [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with
    `./isaaclab.sh --format`
    - [ ] I have made corresponding changes to the documentation
    - [x] My changes generate no new warnings
    - [x] I have added tests that prove my fix is effective or that my
    feature works
    - [ ] I have updated the changelog and the corresponding version in the
    extension's `config/extension.toml` file
    - [x] I have added my name to the `CONTRIBUTORS.md` or my name already
    exists there
    3b8da1d3
Name
Last commit
Last update
.aws Loading commit data...
.github Loading commit data...
.vscode Loading commit data...
apps Loading commit data...
docker Loading commit data...
docs Loading commit data...
scripts Loading commit data...
source Loading commit data...
tools Loading commit data...
.dockerignore Loading commit data...
.flake8 Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.pre-commit-config.yaml Loading commit data...
CONTRIBUTING.md Loading commit data...
CONTRIBUTORS.md Loading commit data...
LICENSE Loading commit data...
LICENSE-mimic Loading commit data...
README.md Loading commit data...
SECURITY.md Loading commit data...
VERSION Loading commit data...
environment.yml Loading commit data...
isaaclab.bat Loading commit data...
isaaclab.sh Loading commit data...
pyproject.toml Loading commit data...