"vscode:/vscode.git/clone" did not exist on "c95080002d87eb722f245f5e5766b2f7dd20a4c7"
Unverified Commit e90582f2 authored by Oleg Goncharov's avatar Oleg Goncharov Committed by GitHub
Browse files

[Common] Removed activations from NVFP4 quantize C++ unit tests (#2289)



* Removed activations from NVFP4 CPP tests. Removed CMake debugging flags
Signed-off-by: default avatarOleg Goncharov <ogoncharov@nvidia.com>

* Better wording
Signed-off-by: default avatarOleg Goncharov <ogoncharov@nvidia.com>

---------
Signed-off-by: default avatarOleg Goncharov <ogoncharov@nvidia.com>
parent b4a1d4d6
...@@ -32,12 +32,6 @@ add_executable(test_operator ...@@ -32,12 +32,6 @@ add_executable(test_operator
test_swap_first_dims.cu test_swap_first_dims.cu
../test_common.cu) ../test_common.cu)
# Add profiling and debug flags for CUDA compilation
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -lineinfo") # Generate line info for device code
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -g") # Add debug symbols for host code
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --ptxas-options=-v") # Add info about registers usage
# Note: Using -lineinfo instead of -G to avoid conflicts and get line mapping
# Find required packages # Find required packages
find_package(OpenMP REQUIRED) find_package(OpenMP REQUIRED)
list(APPEND test_operator_LINKER_LIBS CUDA::cudart GTest::gtest_main ${TE_LIB} CUDA::nvrtc CUDNN::cudnn) list(APPEND test_operator_LINKER_LIBS CUDA::cudart GTest::gtest_main ${TE_LIB} CUDA::nvrtc CUDNN::cudnn)
......
...@@ -661,14 +661,9 @@ std::vector<std::vector<size_t>> tensor_dims = { ...@@ -661,14 +661,9 @@ std::vector<std::vector<size_t>> tensor_dims = {
{4096, 13312}, {4096, 13312},
}; };
// Only GeLU activation tests are supported // Only the Identity activation is currently supported.
std::vector<ActivationType> Activation_types = { std::vector<ActivationType> Activation_types = {
ActivationType::Identity, ActivationType::Identity
ActivationType::GeLU,
ActivationType::SiLU,
ActivationType::ReLU,
ActivationType::QGeLU,
ActivationType::SReLU,
}; };
} // namespace } // namespace
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment