changes made for testing

17b72bd6 · Astha Rai · edac496b · 17b72bd6
Commit 17b72bd6 authored Sep 13, 2023 by Astha Rai
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

example/44_elementwise_permute/elementwise_permute_5D_3d.cpp example/44_elementwise_permute/elementwise_permute_5D_3d.cpp +4 -1

No files found.
--- a/example/44_elementwise_permute/elementwise_permute_5D_3d.cpp
+++ b/example/44_elementwise_permute/elementwise_permute_5D_3d.cpp
@@ -61,7 +61,10 @@ int main()
    Tensor<ADataType> a(ncdhw);
    Tensor<BDataType> b(nchwd);

-    a.GenerateTensorValue(GeneratorTensor_3<ADataType>{0.0, 1.0});
+    //a.GenerateTensorValue(GeneratorTensor_3<ADataType>{0.0, 1.0});
+    for(std::size_t i = 0; i < a.mData.size(); i++){
+	    	    a.mData[i] = i;
+    }

    DeviceMem a_device_buf(sizeof(ADataType) * a.mDesc.GetElementSpaceSize());
    DeviceMem b_device_buf(sizeof(BDataType) * b.mDesc.GetElementSpaceSize());