Expand unit tests for different stride options and malloc_device testing

[mmeterel]

Summary

Expand unit tests to cover different stride options (i.e. stride=0) where applicable.

Expand unit tests to cover malloc device testing.

Problem statement

These additional tests will improve the stability and test corner cases.