fix format and rebase dev

Author: qinyiqun

include/data_type.h

@@ -46,12 +46,4 @@ const static struct DataLayout
     F64  = {1, 1, 8, 52, 11};
 // clang-format on
 
-DT get_F16();
-
-DT get_U32();
-
-DT get_F32();
-
-DT get_U64();
-
 #endif// __DATA_TYPE_H__

operatorspy/tests/random_sample.py

@@ -94,7 +94,7 @@ def test(lib, handle, torch_device, voc, random_val, topp, topk, temperature, x_
     if(torch_device == 'maca'):
         indices = torch.zeros([1], dtype = torch.int64).to('cuda')
     else:
-        indices = torch.zeros([1], dtype = torch.uint64).to(torch_device)
+        indices = torch.zeros([1], dtype = torch.int64).to(torch_device)
     x_tensor = to_tensor(data, lib)
     indices_tensor = to_tensor(indices, lib)
     indices_tensor.descriptor.contents.dt = U64  # treat int64 as uint64

operatorspy/tests/rotary_embedding.py

@@ -245,4 +245,4 @@ def test_musa(lib, test_cases) :
         test_musa(lib, test_cases)
     if not (args.cpu or args.cuda or args.bang or args.ascend or args.maca or args.musa):
         test_cpu(lib, test_cases)
-    print("\033[92mTest passed!\033[0m")
+    print("\033[92mTest passed!\033[0m")

src/devices/musa/common_musa.h

@@ -74,4 +74,4 @@ inline __device__ uint64_t getOffset(uint64_t flat_index, uint64_t ndim, uint64_
     return res;
 }
 
-#endif // __COMMON_MUSA_H__
+#endif // __COMMON_MUSA_H__

src/devices/musa/musa_handle.cc

@@ -16,7 +16,7 @@ infiniopStatus_t createMusaHandle(MusaHandle_t* handle_ptr, int device_id) {
         return STATUS_BAD_DEVICE;
     }
 
-    // set CUDA device property
+    // set MUSA device property
     musaDeviceProp prop;
     musaGetDeviceProperties(&prop, device_id);
 
@@ -54,4 +54,4 @@ infiniopStatus_t deleteMusaHandle(MusaHandle_t handle_ptr) {
     delete handle_ptr;
 
     return STATUS_SUCCESS;
-}
+}

src/devices/musa/musa_handle.h

@@ -61,4 +61,4 @@ void use_mudnn(std::shared_ptr<Pool<musa::dnn::Handle>> mudnn_handles_t, int dev
     mudnn_handles_t->push(handle);
 }
 
-#endif // __MUSA_HANDLE_H__
+#endif // __MUSA_HANDLE_H__

src/devices/musa/pool.h

@@ -47,4 +47,4 @@ class Pool {
     mutable std::atomic<Node<T> *> _head;
 };
 
-#endif // __POOL_MUSA_H__
+#endif // __POOL_MUSA_H__

src/devices/musa/tensor_desc.cc

@@ -32,5 +32,4 @@ infiniopStatus_t musaCausalSoftmax(CausalSoftmaxMusaDescriptor_t desc,
                                    void *stream);
 
 infiniopStatus_t musaDestroyCausalSoftmaxDescriptor(CausalSoftmaxMusaDescriptor_t desc);
-
-#endif
+#endif