Torch integration

CMakeLists.txt

@@ -191,6 +191,11 @@ if(MSCCLPP_NPKIT_FLAGS)
     target_compile_definitions(mscclpp_obj PRIVATE ${MSCCLPP_NPKIT_FLAGS})
 endif()
 
+if(MSCCLPP_USE_ROCM)
+    file(GLOB_RECURSE CU_SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/src/*.cu)
+    set_source_files_properties(${CU_SOURCES} PROPERTIES LANGUAGE CXX)
+endif()
+
 # libmscclpp
 add_library(mscclpp SHARED)
 target_link_libraries(mscclpp PUBLIC mscclpp_obj)